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

Switch stable diffusion to the new tuned model (#455)

Browse Source
This commit is contained in:
Quinn Dawkins
2022-10-31 18:15:31 -04:00
committed by GitHub
parent a3fdfc81db
commit 3e7efaa048
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/main.py
View File

@@ -22,7 +22,7 @@ UNET_FP32 = "unet_fp32"
IREE_EXTRA_ARGS = []
TUNED_GCLOUD_BUCKET = "gs://shark_tank/quinn"
UNET_FP16_TUNED = "unet_fp16_tuned"
UNET_FP16_TUNED = "unet_fp16_tunedv2"
BATCH_SIZE = len(args.prompts)