extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-02-17 02:40:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

Allow the user to specify any disk path to the model, in the config

Browse Source
This commit is contained in:
cmdr2 2022-10-27 23:39:29 +05:30
parent 389e3397ec
commit 9ad1e0d529
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
ui/server.py
View File

@ -141,10 +141,10 @@ def resolve_model_to_use(model_name:str=None):
models_dir_path = os.path.join(MODELS_DIR, 'stable-diffusion', model_name)
if os.path.exists(models_dir_path + '.ckpt'):
return models_dir_path
models_direct_dir_path = os.path.join(SD_DIR, model_name)
if os.path.exists(models_direct_dir_path + '.ckpt'):
if os.path.exists(model_name + '.ckpt'):
# Direct Path to file
return models_direct_dir_path
model_name = os.path.abspath(model_name)
return model_name
# Default locations
if model_name in APP_CONFIG_DEFAULT_MODELS:
default_model_path = os.path.join(SD_DIR, model_name)