ok

hydra_node/models.py

@@ -194,7 +194,7 @@ class StableDiffusionModel(nn.Module):
         self.config = config
         self.premodules = None
         if self.config.use_s3 == '1':
-            self.model = self.from_url(self.config.model_path)
+            model, model_config = self.from_url(self.config.model_path)
 
         else:
             if Path(self.config.model_path).is_dir():