disable use_tuned on SD1.4 for rdna2 (#1490)

this is a temporary measure while we retune SD1.4 for rdna2. The current config fails during iree-compile.
This commit is contained in:
Ean Garvey
2023-06-05 19:46:16 -05:00
committed by GitHub
parent 8400529c2c
commit d0ba3ef8fa

View File

@@ -342,6 +342,15 @@ def set_init_device_flags():
):
args.use_tuned = False
elif "rdna2" in args.iree_vulkan_target_triple and (
base_model_id
not in [
"stabilityai/stable-diffusion-2-1",
"stabilityai/stable-diffusion-2-1-base",
]
):
args.use_tuned = False
if args.use_tuned:
print(
f"Using tuned models for {base_model_id}(fp16) on device {args.device}."