extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2024-11-22 08:13:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Allow using a custom ckpt model

Browse Source
This commit is contained in:
cmdr2 2022-09-28 15:10:05 +05:30
parent 553525a0fb
commit 2696da7337
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ui/server.py
View File

@ -84,7 +84,10 @@ async def ping():
model_is_loading = True
from sd_internal import runtime
runtime.load_model_ckpt(ckpt_to_use="sd-v1-4")
custom_weight_path = os.path.join(SCRIPT_DIR, 'custom-model.ckpt')
ckpt_to_use = "sd-v1-4" if not os.path.exists(custom_weight_path) else "custom-model"
runtime.load_model_ckpt(ckpt_to_use=ckpt_to_use)
model_loaded = True
model_is_loading = False