gah

hydra_node/config.py

@@ -140,6 +140,7 @@ def init_config_model():
     logger.info(f"MODEL: {config.model_name}")
     
     #S3 auth stuff
+    config.use_s3 = os.getenv('USE_S3', '0')
     config.s3_access_key = os.getenv('S3_ACCESS_KEY', None)
     config.s3_secret_key = os.getenv('S3_SECRET_KEY', None)
     config.s3_bucket = os.getenv('S3_BUCKET', None)

hydra_node/models.py

@@ -193,7 +193,7 @@ class StableDiffusionModel(nn.Module):
         nn.Module.__init__(self)
         self.config = config
         self.premodules = None
-        if self.config.model_path.startswith("https://"):
+        if self.config.use_s3 == '1':
             self.model = self.from_url(self.config.model_path)
 
         else: