erroring

hydra_node/config.py

@@ -146,12 +146,6 @@ def init_config_model():
     config.s3_file = os.environ('S3_FILE', None)
     config.s3_endpoint = os.environ('S3_ENDPOINT', None)
 
-    try:
-        config.model_path = f"https://{config.s3_endpoint}/{config.s3_bucket}/{config.s3_file}"
-        logger.info(f"Path is set to S3 {config.model_path}")
-    except:
-        pass
-
     # Resolve where we get our model and data from.
     config.model_path = os.getenv('MODEL_PATH', None)
     config.enable_ema = os.getenv('ENABLE_EMA', "1")
@@ -178,6 +172,14 @@ def init_config_model():
     # Instantiate our actual model.
     load_time = time.time()
     model_hash = None
+
+    if config.model_path is None:
+        try:
+            config.model_path = f"https://{config.s3_endpoint}/{config.s3_bucket}/{config.s3_file}"
+            logger.info(f"Path is set to S3 {config.model_path}")
+        except:
+            logger.error("No model path or S3 info provided")
+            raise
     
     try:
         if config.model_name != "dalle-mini":