mirror of
https://github.com/lllyasviel/Fooocus.git
synced 2026-08-16 13:13:16 +02:00
i
This commit is contained in:
@@ -985,7 +985,7 @@ class UNetModel(nn.Module):
|
|||||||
|
|
||||||
if self.num_classes is not None:
|
if self.num_classes is not None:
|
||||||
assert y.shape[0] == x.shape[0]
|
assert y.shape[0] == x.shape[0]
|
||||||
emb = emb + self.label_emb(y)
|
emb = emb + self.label_emb(y.to(x))
|
||||||
|
|
||||||
# h = x.type(self.dtype)
|
# h = x.type(self.dtype)
|
||||||
h = x
|
h = x
|
||||||
|
|||||||
Reference in New Issue
Block a user