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

nod-ai · Jul 1, 2023 · 6d286c0 · 6d286c0
1 parent 7392b22
commit 6d286c0
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/apps/stable_diffusion/src/utils/utils.py b/apps/stable_diffusion/src/utils/utils.py
@@ -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