diff --git a/iree_tests/download_remote_files.py b/iree_tests/download_remote_files.py
index f62fc62b4..225764f05 100644
--- a/iree_tests/download_remote_files.py
+++ b/iree_tests/download_remote_files.py
@@ -98,6 +98,8 @@ def download_azure_remote_file(test_dir: Path, remote_file: str):
         with open(local_file_path, mode="wb") as local_blob:
             download_stream = blob_client.download_blob(max_concurrency=4)
             local_blob.write(download_stream.readall())
+        if cache_location != REPO_ROOT:
+            os.symlink(local_file_path, test_dir / remote_file_name)
 
 
 def download_generic_remote_file(test_dir: Path, remote_file: str):
diff --git a/iree_tests/future-pytorch-models/llama-tank/real_weights_data_flags.txt b/iree_tests/future-pytorch-models/llama-tank/real_weights_data_flags.txt
index 86ff2b8ad..327fcb28d 100644
--- a/iree_tests/future-pytorch-models/llama-tank/real_weights_data_flags.txt
+++ b/iree_tests/future-pytorch-models/llama-tank/real_weights_data_flags.txt
@@ -1,67 +1,67 @@
---parameters=${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/real_weights.irpa
---input=1x93xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_input.0.bin
---expected_output=1x93x32000xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.0.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.1.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.2.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.3.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.4.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.5.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.6.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.7.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.8.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.9.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.10.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.11.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.12.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.13.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.14.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.15.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.16.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.17.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.18.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.19.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.20.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.21.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.22.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.23.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.24.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.25.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.26.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.27.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.28.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.29.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.30.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.31.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.32.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.33.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.34.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.35.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.36.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.37.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.38.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.39.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.40.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.41.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.42.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.43.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.44.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.45.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.46.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.47.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.48.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.49.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.50.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.51.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.52.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.53.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.54.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.55.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.56.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.57.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.58.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.59.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.60.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.61.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.62.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.63.bin
---expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.64.bin
+--parameters=real_weights.irpa
+--input=1x93xi64=@inference_input.0.bin
+--expected_output=1x93x32000xf32=@inference_output.0.bin
+--expected_output=1x32x93x128xf32=@inference_output.1.bin
+--expected_output=1x32x93x128xf32=@inference_output.2.bin
+--expected_output=1x32x93x128xf32=@inference_output.3.bin
+--expected_output=1x32x93x128xf32=@inference_output.4.bin
+--expected_output=1x32x93x128xf32=@inference_output.5.bin
+--expected_output=1x32x93x128xf32=@inference_output.6.bin
+--expected_output=1x32x93x128xf32=@inference_output.7.bin
+--expected_output=1x32x93x128xf32=@inference_output.8.bin
+--expected_output=1x32x93x128xf32=@inference_output.9.bin
+--expected_output=1x32x93x128xf32=@inference_output.10.bin
+--expected_output=1x32x93x128xf32=@inference_output.11.bin
+--expected_output=1x32x93x128xf32=@inference_output.12.bin
+--expected_output=1x32x93x128xf32=@inference_output.13.bin
+--expected_output=1x32x93x128xf32=@inference_output.14.bin
+--expected_output=1x32x93x128xf32=@inference_output.15.bin
+--expected_output=1x32x93x128xf32=@inference_output.16.bin
+--expected_output=1x32x93x128xf32=@inference_output.17.bin
+--expected_output=1x32x93x128xf32=@inference_output.18.bin
+--expected_output=1x32x93x128xf32=@inference_output.19.bin
+--expected_output=1x32x93x128xf32=@inference_output.20.bin
+--expected_output=1x32x93x128xf32=@inference_output.21.bin
+--expected_output=1x32x93x128xf32=@inference_output.22.bin
+--expected_output=1x32x93x128xf32=@inference_output.23.bin
+--expected_output=1x32x93x128xf32=@inference_output.24.bin
+--expected_output=1x32x93x128xf32=@inference_output.25.bin
+--expected_output=1x32x93x128xf32=@inference_output.26.bin
+--expected_output=1x32x93x128xf32=@inference_output.27.bin
+--expected_output=1x32x93x128xf32=@inference_output.28.bin
+--expected_output=1x32x93x128xf32=@inference_output.29.bin
+--expected_output=1x32x93x128xf32=@inference_output.30.bin
+--expected_output=1x32x93x128xf32=@inference_output.31.bin
+--expected_output=1x32x93x128xf32=@inference_output.32.bin
+--expected_output=1x32x93x128xf32=@inference_output.33.bin
+--expected_output=1x32x93x128xf32=@inference_output.34.bin
+--expected_output=1x32x93x128xf32=@inference_output.35.bin
+--expected_output=1x32x93x128xf32=@inference_output.36.bin
+--expected_output=1x32x93x128xf32=@inference_output.37.bin
+--expected_output=1x32x93x128xf32=@inference_output.38.bin
+--expected_output=1x32x93x128xf32=@inference_output.39.bin
+--expected_output=1x32x93x128xf32=@inference_output.40.bin
+--expected_output=1x32x93x128xf32=@inference_output.41.bin
+--expected_output=1x32x93x128xf32=@inference_output.42.bin
+--expected_output=1x32x93x128xf32=@inference_output.43.bin
+--expected_output=1x32x93x128xf32=@inference_output.44.bin
+--expected_output=1x32x93x128xf32=@inference_output.45.bin
+--expected_output=1x32x93x128xf32=@inference_output.46.bin
+--expected_output=1x32x93x128xf32=@inference_output.47.bin
+--expected_output=1x32x93x128xf32=@inference_output.48.bin
+--expected_output=1x32x93x128xf32=@inference_output.49.bin
+--expected_output=1x32x93x128xf32=@inference_output.50.bin
+--expected_output=1x32x93x128xf32=@inference_output.51.bin
+--expected_output=1x32x93x128xf32=@inference_output.52.bin
+--expected_output=1x32x93x128xf32=@inference_output.53.bin
+--expected_output=1x32x93x128xf32=@inference_output.54.bin
+--expected_output=1x32x93x128xf32=@inference_output.55.bin
+--expected_output=1x32x93x128xf32=@inference_output.56.bin
+--expected_output=1x32x93x128xf32=@inference_output.57.bin
+--expected_output=1x32x93x128xf32=@inference_output.58.bin
+--expected_output=1x32x93x128xf32=@inference_output.59.bin
+--expected_output=1x32x93x128xf32=@inference_output.60.bin
+--expected_output=1x32x93x128xf32=@inference_output.61.bin
+--expected_output=1x32x93x128xf32=@inference_output.62.bin
+--expected_output=1x32x93x128xf32=@inference_output.63.bin
+--expected_output=1x32x93x128xf32=@inference_output.64.bin
diff --git a/iree_tests/future-pytorch-models/sd-clip-tank/real_weights_data_flags.txt b/iree_tests/future-pytorch-models/sd-clip-tank/real_weights_data_flags.txt
index 4b591c14c..b5fe5e578 100644
--- a/iree_tests/future-pytorch-models/sd-clip-tank/real_weights_data_flags.txt
+++ b/iree_tests/future-pytorch-models/sd-clip-tank/real_weights_data_flags.txt
@@ -1,4 +1,4 @@
---parameters=model=${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-clip-tank/real_weights.irpa
---input=1x77xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-clip-tank/inference_input.0.bin
---expected_output=1x77x768xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-clip-tank/inference_output.0.bin
---expected_output=1x768xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-clip-tank/inference_output.1.bin
+--parameters=model=real_weights.irpa
+--input=1x77xi64=@inference_input.0.bin
+--expected_output=1x77x768xf32=@inference_output.0.bin
+--expected_output=1x768xf32=@inference_output.1.bin
diff --git a/iree_tests/future-pytorch-models/sd-unet-tank/real_weights_data_flags.txt b/iree_tests/future-pytorch-models/sd-unet-tank/real_weights_data_flags.txt
index 4a92c5bae..cb3c458ca 100644
--- a/iree_tests/future-pytorch-models/sd-unet-tank/real_weights_data_flags.txt
+++ b/iree_tests/future-pytorch-models/sd-unet-tank/real_weights_data_flags.txt
@@ -1,5 +1,5 @@
---parameters=model=${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-unet-tank/real_weights.irpa
---input=1x4x64x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-unet-tank/inference_input.0.bin
---input=1xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-unet-tank/inference_input.1.bin
---input=2x77x768xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-unet-tank/inference_input.2.bin
---expected_output=1x4x64x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-unet-tank/inference_output.0.bin
+--parameters=model=real_weights.irpa
+--input=1x4x64x64xf32=@inference_input.0.bin
+--input=1xf32=@inference_input.1.bin
+--input=2x77x768xf32=@inference_input.2.bin
+--expected_output=1x4x64x64xf32=@inference_output.0.bin
diff --git a/iree_tests/future-pytorch-models/sd-vae-decode-tank/real_weights_data_flags.txt b/iree_tests/future-pytorch-models/sd-vae-decode-tank/real_weights_data_flags.txt
index 49c5e8884..6b90771ec 100644
--- a/iree_tests/future-pytorch-models/sd-vae-decode-tank/real_weights_data_flags.txt
+++ b/iree_tests/future-pytorch-models/sd-vae-decode-tank/real_weights_data_flags.txt
@@ -1,3 +1,3 @@
---parameters=model=${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-vae-decode-tank/real_weights.irpa
---input=1x4x64x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-vae-decode-tank/inference_input.0.bin
---expected_output=1x3x512x512xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-vae-decode-tank/inference_output.0.bin
+--parameters=model=real_weights.irpa
+--input=1x4x64x64xf32=@inference_input.0.bin
+--expected_output=1x3x512x512xf32=@inference_output.0.bin
diff --git a/iree_tests/pytorch/models/opt-125M/real_weights_data_flags.txt b/iree_tests/pytorch/models/opt-125M/real_weights_data_flags.txt
index 0d1646312..49835b598 100644
--- a/iree_tests/pytorch/models/opt-125M/real_weights_data_flags.txt
+++ b/iree_tests/pytorch/models/opt-125M/real_weights_data_flags.txt
@@ -1,27 +1,27 @@
---parameters=${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/real_weights.irpa
---input=1x8xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_input.0.bin
---expected_output=1x8x50272xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.0.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.1.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.2.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.3.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.4.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.5.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.6.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.7.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.8.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.9.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.10.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.11.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.12.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.13.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.14.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.15.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.16.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.17.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.18.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.19.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.20.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.21.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.22.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.23.bin
---expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.24.bin
+--parameters=real_weights.irpa
+--input=1x8xi64=@inference_input.0.bin
+--expected_output=1x8x50272xf32=@inference_output.0.bin
+--expected_output=1x12x8x64xf32=@inference_output.1.bin
+--expected_output=1x12x8x64xf32=@inference_output.2.bin
+--expected_output=1x12x8x64xf32=@inference_output.3.bin
+--expected_output=1x12x8x64xf32=@inference_output.4.bin
+--expected_output=1x12x8x64xf32=@inference_output.5.bin
+--expected_output=1x12x8x64xf32=@inference_output.6.bin
+--expected_output=1x12x8x64xf32=@inference_output.7.bin
+--expected_output=1x12x8x64xf32=@inference_output.8.bin
+--expected_output=1x12x8x64xf32=@inference_output.9.bin
+--expected_output=1x12x8x64xf32=@inference_output.10.bin
+--expected_output=1x12x8x64xf32=@inference_output.11.bin
+--expected_output=1x12x8x64xf32=@inference_output.12.bin
+--expected_output=1x12x8x64xf32=@inference_output.13.bin
+--expected_output=1x12x8x64xf32=@inference_output.14.bin
+--expected_output=1x12x8x64xf32=@inference_output.15.bin
+--expected_output=1x12x8x64xf32=@inference_output.16.bin
+--expected_output=1x12x8x64xf32=@inference_output.17.bin
+--expected_output=1x12x8x64xf32=@inference_output.18.bin
+--expected_output=1x12x8x64xf32=@inference_output.19.bin
+--expected_output=1x12x8x64xf32=@inference_output.20.bin
+--expected_output=1x12x8x64xf32=@inference_output.21.bin
+--expected_output=1x12x8x64xf32=@inference_output.22.bin
+--expected_output=1x12x8x64xf32=@inference_output.23.bin
+--expected_output=1x12x8x64xf32=@inference_output.24.bin
diff --git a/iree_tests/pytorch/models/resnet50/real_weights_data_flags.txt b/iree_tests/pytorch/models/resnet50/real_weights_data_flags.txt
index 69c24f0ce..5aae46217 100644
--- a/iree_tests/pytorch/models/resnet50/real_weights_data_flags.txt
+++ b/iree_tests/pytorch/models/resnet50/real_weights_data_flags.txt
@@ -1,3 +1,3 @@
---parameters=${IREE_TEST_FILES}/iree_tests/pytorch/models/resnet50/real_weights.irpa
---input=1x3x224x224xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/resnet50/inference_input.0.bin
---expected_output=1x1000xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/resnet50/inference_output.0.bin
+--parameters=real_weights.irpa
+--input=1x3x224x224xf32=@inference_input.0.bin
+--expected_output=1x1000xf32=@inference_output.0.bin
diff --git a/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/real_weights_data_flags.txt b/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/real_weights_data_flags.txt
index 1934bafd6..fc85e7190 100644
--- a/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/real_weights_data_flags.txt
+++ b/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/real_weights_data_flags.txt
@@ -1,8 +1,8 @@
---parameters=model=${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/real_weights.irpa
---input=1x64xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/inference_input.0.bin
---input=1x64xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/inference_input.1.bin
---input=1x64xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/inference_input.2.bin
---input=1x64xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/inference_input.3.bin
---expected_output=2x64x2048xf16=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/inference_output.0.bin
---expected_output=2x1280xf16=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/inference_output.1.bin
+--parameters=model=real_weights.irpa
+--input=1x64xi64=@inference_input.0.bin
+--input=1x64xi64=@inference_input.1.bin
+--input=1x64xi64=@inference_input.2.bin
+--input=1x64xi64=@inference_input.3.bin
+--expected_output=2x64x2048xf16=@inference_output.0.bin
+--expected_output=2x1280xf16=@inference_output.1.bin
 --expected_f16_threshold=1.0f
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 a4198a1e3..f6ceb4edd 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,8 @@
---parameters=model=${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-scheduled-unet-3-tank/real_weights.irpa
+--parameters=model=real_weights.irpa
 --module=sdxl_scheduled_unet_pipeline_fp16_.vmfb
---input=1x4x128x128xf16=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-scheduled-unet-3-tank/inference_input.0.bin
---input=2x64x2048xf16=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-scheduled-unet-3-tank/inference_input.1.bin
---input=2x1280xf16=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-scheduled-unet-3-tank/inference_input.2.bin
---input=1xf16=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-scheduled-unet-3-tank/inference_input.3.bin
---expected_output=1x4x128x128xf16=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-scheduled-unet-3-tank/inference_output.0.bin
+--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
diff --git a/iree_tests/pytorch/models/sdxl-vae-decode-tank/real_weights_data_flags.txt b/iree_tests/pytorch/models/sdxl-vae-decode-tank/real_weights_data_flags.txt
index 608710223..aceb8839c 100644
--- a/iree_tests/pytorch/models/sdxl-vae-decode-tank/real_weights_data_flags.txt
+++ b/iree_tests/pytorch/models/sdxl-vae-decode-tank/real_weights_data_flags.txt
@@ -1,4 +1,4 @@
---parameters=model=${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-vae-decode-tank/real_weights.irpa
---input=1x4x128x128xf16=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-vae-decode-tank/inference_input.0.bin
---expected_output=1x3x1024x1024xf16=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-vae-decode-tank/inference_output.0.bin
+--parameters=model=real_weights.irpa
+--input=1x4x128x128xf16=@inference_input.0.bin
+--expected_output=1x3x1024x1024xf16=@inference_output.0.bin
 --expected_f16_threshold=0.02f