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

Enable v1.5 to use tuned configs (#1049)

Browse Source
This commit is contained in:
yzhang93
2023-02-18 13:54:26 -08:00
committed by GitHub
parent 962470f610
commit 5b33597b6d
2 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
apps/stable_diffusion/src/utils/sd_annotation.py
View File

@@ -85,6 +85,8 @@ def load_lower_configs():
base_model_id = args.hf_model_id
if args.ckpt_loc != "":
base_model_id = fetch_and_update_base_model_id(args.ckpt_loc)
if base_model_id == "runwayml/stable-diffusion-v1-5":
base_model_id = "CompVis/stable-diffusion-v1-4"
variant, version = get_variant_version(base_model_id)

1
apps/stable_diffusion/src/utils/utils.py
View File

@@ -265,6 +265,7 @@ def set_init_device_flags():
"stabilityai/stable-diffusion-2-1",
"stabilityai/stable-diffusion-2-1-base",
"CompVis/stable-diffusion-v1-4",
"runwayml/stable-diffusion-v1-5",
]:
args.use_tuned = False