no_init breaks dalle-mini lol, how funky!

hydra_node/config.py

@@ -99,7 +99,11 @@ def init_config_model():
     load_time = time.time()
 
     try:
-        model = no_init(lambda: model_map[config.model_name](config))
+        if config.model_name != "dalle-mini":
+            model = no_init(lambda: model_map[config.model_name](config))
+        else:
+            model = model_map[config.model_name](config)
+            
     except Exception as e:
         logger.error(f"Failed to load model: {str(e)}")
         capture_exception(e)

run.sh

 export DTYPE="float16"
-export MODEL="test"
+export MODEL="dalle-mini"
 export DEV="True"
-export MODEL_PATH="/home/xuser/diffusionstorage/workspace/kuru/stablediff/v1pp-flatline-pruned"
+export MODEL_PATH="/home/xuser/diffusionstorage/workspace/kuru/stablediff/dalle-mini-models"
 export SENTRY_URL="https://49ca8adcf4444f82a10eae1b3fd4182f@o846434.ingest.sentry.io/6612448"
 gunicorn main:app --workers 1 --worker-class uvicorn.workers.UvicornWorker --bind 0.0.0.0:80
\ No newline at end of file