diff --git a/iree_tests/pytorch/models/sdxl-scheduled-unet-3-tank/real_weights_data_flags.txt b/iree_tests/pytorch/models/sdxl-scheduled-unet-3-tank/real_weights_data_flags.txt
index d98f8d6df..d3f5a12fa 100644
--- a/iree_tests/pytorch/models/sdxl-scheduled-unet-3-tank/real_weights_data_flags.txt
+++ b/iree_tests/pytorch/models/sdxl-scheduled-unet-3-tank/real_weights_data_flags.txt
@@ -1,8 +1 @@
---parameters=model=real_weights.irpa
---module=sdxl_scheduled_unet_pipeline_fp16_.vmfb
---input=1x4x128x128xf16=@inference_input.0.bin
---input=2x64x2048xf16=@inference_input.1.bin
---input=2x1280xf16=@inference_input.2.bin
---input=1xf16=@inference_input.3.bin
---expected_output=1x4x128x128xf16=@inference_output.0.bin
---expected_f16_threshold=0.8f
+