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

Update opt_params.py

Browse Source
This commit is contained in:
Prashant Kumar
2022-12-19 23:39:04 +05:30
committed by GitHub
parent a17800da00
commit b466b51247
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
shark/examples/shark_inference/stable_diffusion/opt_params.py
View File

@@ -114,10 +114,7 @@ def get_vae():
bucket = "gs://shark_tank/stable_diffusion"
model_name = "vae_19dec_fp16"
if args.version == "v2.1base":
if args.max_length == 64:
model_name = "vae_19dec_v2p1base_fp16_64"
else:
model_name = "vae2base_19dec_fp16"
model_name = "vae2base_19dec_fp16"
if args.version == "v2.1":
model_name = "vae2_19dec_fp16"
iree_flags += [