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

[SD][web] Set default tuned unet to v2 (#663)

Browse Source
This commit is contained in:
Quinn Dawkins
2022-12-18 23:50:08 -05:00
committed by GitHub
parent 59358361f9
commit 7d0cbd8d90
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
web/models/stable_diffusion/opt_params.py
View File

@@ -29,7 +29,7 @@ def get_unet():
if args.version == "v1.4":
model_name = "unet_1dec_fp16_tuned"
if args.version == "v2.1base":
model_name = "unet2base_8dec_fp16_tuned"
model_name = "unet2base_8dec_fp16_tuned_v2"
return get_shark_model(bucket, model_name, iree_flags)
else:
bucket = "gs://shark_tank/stable_diffusion"