From 8807b2dff8de2c4a9f7554453314ba89cb2f5488 Mon Sep 17 00:00:00 2001 From: lvmin Date: Wed, 9 Aug 2023 14:02:43 -0700 Subject: [PATCH] i --- sgm/modules/diffusionmodules/openaimodel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sgm/modules/diffusionmodules/openaimodel.py b/sgm/modules/diffusionmodules/openaimodel.py index ce16fd21..1e81752b 100644 --- a/sgm/modules/diffusionmodules/openaimodel.py +++ b/sgm/modules/diffusionmodules/openaimodel.py @@ -985,7 +985,7 @@ class UNetModel(nn.Module): if self.num_classes is not None: 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