gate it

hydra_node/config.py

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

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