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

Disable vae --use_tuned version.

Browse Source
This commit is contained in:
Prashant Kumar
2022-12-19 17:11:26 +00:00
parent 9a36816d27
commit 059c1b3a19
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -96,7 +96,7 @@ def get_vae():
if sys.platform == "darwin":
iree_flags.append("-iree-stream-fuse-binding=false")
if args.precision in ["fp16", "int8"]:
if args.use_tuned:
if False:
bucket = "gs://shark_tank/vivian"
if args.version == "v2.1base":
model_name = "vae2base_8dec_fp16_tuned"