fix vectoradjust

hydra_node/models.py

@@ -169,7 +169,7 @@ class StableDiffusionModel(nn.Module):
             'k_lms': K.sampling.sample_lms,
         }
         if config.prior_path:
-            self.prior = VectorAdjustPrior.load_model(config.prior_path)
+            self.prior = VectorAdjustPrior.load_model(config.prior_path).to(self.device)
 
     def from_folder(self, folder):
         folder = Path(folder)