i

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