From c1798a3728e955c06b51796e17b4de7469487e81 Mon Sep 17 00:00:00 2001 From: saienduri <77521230+saienduri@users.noreply.github.com> Date: Wed, 24 Apr 2024 12:30:27 -0700 Subject: [PATCH] add back real_weights_data_flags.txt --- .../real_weights_data_flags.txt | 8 ++++++++ 1 file changed, 8 insertions(+) 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 e69de29bb..d98f8d6df 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 @@ -0,0 +1,8 @@ +--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