Update real_weights_data_flags.txt

nod-ai · Apr 24, 2024 · 88f3c99 · 88f3c99
1 parent 22e76b4
commit 88f3c99
Showing 1 changed file with 1 addition and 8 deletions.
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
@@ -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
+