fix bug

hydra_node/models.py

@@ -134,6 +134,7 @@ class StableDiffusionModel(nn.Module):
             typex = torch.float16
         else:
             typex = torch.float32
+        self.model = model.to(config.device).to(typex)
         self.k_model = K.external.CompVisDenoiser(model)
         self.k_model = StableInterface(self.k_model)
         self.device = config.device