diff --git a/models/turbine_models/custom_models/sd_inference/utils.py b/models/turbine_models/custom_models/sd_inference/utils.py index 9d5c149a..2b6a164c 100644 --- a/models/turbine_models/custom_models/sd_inference/utils.py +++ b/models/turbine_models/custom_models/sd_inference/utils.py @@ -40,6 +40,7 @@ "--iree-opt-data-tiling=false", "--iree-codegen-gpu-native-math-precision=true", "--iree-vm-target-truncate-unsupported-floats", + "--iree-flow-enable-fuse-horizontal-contractions=true", ], "clip": [ "--iree-flow-enable-aggressive-fusion",