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

nod-ai · Jun 21, 2023 · 1a8e202 · 1a8e202
1 parent d61b664
commit 1a8e202
Showing 1 changed file with 5 additions and 0 deletions.
diff --git 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",