github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[Web] Fix custom model path for upscaler (#1199)

Browse Source
This commit is contained in:
m68k-fr
2023-03-16 23:57:23 +01:00
committed by GitHub
parent 0225434389
commit b661656c03
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
apps/stable_diffusion/scripts/upscaler.py
View File

@@ -78,7 +78,7 @@ def upscaler_inf(
)
args.hf_model_id = hf_model_id
elif ".ckpt" in custom_model or ".safetensors" in custom_model:
args.ckpt_loc = custom_model
args.ckpt_loc = get_custom_model_pathfile(custom_model)
else:
args.hf_model_id = custom_model