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

Pass the flags to vae. (#422)

Browse Source
This commit is contained in:
Prashant Kumar
2022-10-23 23:53:13 +05:30
committed by GitHub
parent 4f906a265c
commit 2741b8be53
1 changed files with 16 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
shark/examples/shark_inference/stable_diffusion/main.py
View File

@@ -26,8 +26,22 @@ def get_models():
if args.import_mlir == True:
return get_vae16(), get_unet16_wrapped()
else:
return get_shark_model(GCLOUD_BUCKET, VAE_FP16), get_shark_model(
GCLOUD_BUCKET, UNET_FP16
return get_shark_model(
GCLOUD_BUCKET,
VAE_FP16,
[
"--iree-flow-enable-conv-nchw-to-nhwc-transform",
"--iree-flow-enable-padding-linalg-ops",
"--iree-flow-linalg-ops-padding-size=16",
],
), get_shark_model(
GCLOUD_BUCKET,
UNET_FP16,
[
"--iree-flow-enable-conv-nchw-to-nhwc-transform",
"--iree-flow-enable-padding-linalg-ops",
"--iree-flow-linalg-ops-padding-size=16",
],
)
elif args.precision == "fp32":