gate it

hydra_node/config.py

@@ -166,9 +166,10 @@ def init_config_model():
         model_hash = crc32(model_path)
 
         #Load Modules
-        modules = load_modules(config.module_path)
-        #attach it to the model
-        model.premodules = modules
+        if config.module_path is not None:
+            modules = load_modules(config.module_path)
+            #attach it to the model
+            model.premodules = modules
 
     config.model = model
 

run_diffusion.sh

@@ -3,6 +3,7 @@ export AMP="1"
 export MODEL="stable-diffusion"
 export DEV="True"
 export MODEL_PATH="/home/xuser/nvme1/stableckpt/v13"
+export MODULE_PATH="/home/xuser/nvme1/stableckpt/modules"
 export TRANSFORMERS_CACHE="/home/xuser/nvme1/transformer_cache"
 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:4315