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

Enable tuning for rectangle sizes on rdna2. (#1608)

Browse Source
This commit is contained in:
Ean Garvey
2023-07-01 00:28:24 -05:00
committed by GitHub
parent 7392b22731
commit 6d286c0609
1 changed files with 7 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
apps/stable_diffusion/src/utils/utils.py
View File

@@ -316,7 +316,13 @@ def set_init_device_flags():
args.use_tuned = False
elif (
args.height != args.width and "rdna2" in args.iree_vulkan_target_triple
args.height != args.width
and "rdna2" in args.iree_vulkan_target_triple
and base_model_id
not in [
"CompVis/stable-diffusion-v1-4",
"runwayml/stable-diffusion-v1-5",
]
):
args.use_tuned = False