Skip to content

H
Hydra Node Http
Commit abbc31df authored by kurumuz's avatar kurumuz
Browse files
Options

probably works now

parent 8e15b74d
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
hydra_node/config.py
View file @ abbc31df
...@@ -187,6 +187,7 @@ def init_config_model(): ...@@ -187,6 +187,7 @@ def init_config_model():
#exit gunicorn #exit gunicorn
sys.exit(4) sys.exit(4)
'''
if config.model_name == "stable-diffusion": if config.model_name == "stable-diffusion":
folder = Path(config.model_path) folder = Path(config.model_path)
if (folder / "pruned.ckpt").is_file(): if (folder / "pruned.ckpt").is_file():
...@@ -200,7 +201,7 @@ def init_config_model(): ...@@ -200,7 +201,7 @@ def init_config_model():
modules = load_modules(config.module_path) modules = load_modules(config.module_path)
#attach it to the model #attach it to the model
model.premodules = modules model.premodules = modules
'''
config.model = model config.model = model
# Mark that our model is loaded. # Mark that our model is loaded.
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment