Exclude non-square sizes from use_tuned on rdna2 (#1568)

2026-01-09 13:57:54 -05:00 · 2023-06-21 11:36:55 -05:00
parent d61b6641fb
commit 1a8e2024d6
1 changed files with 5 additions and 0 deletions
--- a/apps/stable_diffusion/src/utils/utils.py
+++ b/apps/stable_diffusion/src/utils/utils.py
@@ -305,6 +305,11 @@ def set_init_device_flags():
    ):
        args.use_tuned = False

+    elif (
+        args.height != args.width and "rdna2" in args.iree_vulkan_target_triple
+    ):
+        args.use_tuned = False
+
    elif base_model_id not in [
        "Linaqruf/anything-v3.0",
        "dreamlike-art/dreamlike-diffusion-1.0",