[CI Failure] Fix tests with missing TinyLlama-1.1B-Chat-v1.0-FP8-e2e (#26816)

mgoin · web-flow · commit c3a722fcb2c9 · 2025-10-14T18:38:59.000Z
Signed-off-by: mgoin &lt;mgoin64@gmail.com&gt;
diff --git a/tests/compile/test_async_tp.py b/tests/compile/test_async_tp.py
@@ -332,7 +332,7 @@ def async_tp_pass_on_test_model(
 
     # this is a fake model name to construct the model config
     # in the vllm_config, it's not really used.
-    model_name = "nm-testing/TinyLlama-1.1B-Chat-v1.0-FP8-e2e"
+    model_name = "RedHatAI/Llama-3.2-1B-Instruct-FP8"
     vllm_config.model_config = ModelConfig(
         model=model_name, trust_remote_code=True, dtype=dtype, seed=42
     )
diff --git a/tests/compile/test_fusion_all_reduce.py b/tests/compile/test_fusion_all_reduce.py
@@ -229,7 +229,7 @@ def all_reduce_fusion_pass_on_test_model(
 
     # this is a fake model name to construct the model config
     # in the vllm_config, it's not really used.
-    model_name = "nm-testing/TinyLlama-1.1B-Chat-v1.0-FP8-e2e"
+    model_name = "RedHatAI/Llama-3.2-1B-Instruct-FP8"
     vllm_config.model_config = ModelConfig(
         model=model_name, trust_remote_code=True, dtype=dtype, seed=42
     )
diff --git a/tests/compile/test_sequence_parallelism.py b/tests/compile/test_sequence_parallelism.py
@@ -278,7 +278,7 @@ def sequence_parallelism_pass_on_test_model(
 
     # this is a fake model name to construct the model config
     # in the vllm_config, it's not really used.
-    model_name = "nm-testing/TinyLlama-1.1B-Chat-v1.0-FP8-e2e"
+    model_name = "RedHatAI/Llama-3.2-1B-Instruct-FP8"
     vllm_config.model_config = ModelConfig(
         model=model_name, trust_remote_code=True, dtype=dtype, seed=42
     )

Original file line number	Diff line number	Diff line change
`@@ -332,7 +332,7 @@ def async_tp_pass_on_test_model(`
`332`	`332`
`333`	`333`	`# this is a fake model name to construct the model config`
`334`	`334`	`# in the vllm_config, it's not really used.`
`335`		`- model_name = "nm-testing/TinyLlama-1.1B-Chat-v1.0-FP8-e2e"`
	`335`	`+ model_name = "RedHatAI/Llama-3.2-1B-Instruct-FP8"`
`336`	`336`	`vllm_config.model_config = ModelConfig(`
`337`	`337`	`model=model_name, trust_remote_code=True, dtype=dtype, seed=42`
`338`	`338`	`)`
Original file line number	Diff line number	Diff line change
`@@ -229,7 +229,7 @@ def all_reduce_fusion_pass_on_test_model(`
`229`	`229`
`230`	`230`	`# this is a fake model name to construct the model config`
`231`	`231`	`# in the vllm_config, it's not really used.`
`232`		`- model_name = "nm-testing/TinyLlama-1.1B-Chat-v1.0-FP8-e2e"`
	`232`	`+ model_name = "RedHatAI/Llama-3.2-1B-Instruct-FP8"`
`233`	`233`	`vllm_config.model_config = ModelConfig(`
`234`	`234`	`model=model_name, trust_remote_code=True, dtype=dtype, seed=42`
`235`	`235`	`)`
Original file line number	Diff line number	Diff line change
`@@ -278,7 +278,7 @@ def sequence_parallelism_pass_on_test_model(`
`278`	`278`
`279`	`279`	`# this is a fake model name to construct the model config`
`280`	`280`	`# in the vllm_config, it's not really used.`
`281`		`- model_name = "nm-testing/TinyLlama-1.1B-Chat-v1.0-FP8-e2e"`
	`281`	`+ model_name = "RedHatAI/Llama-3.2-1B-Instruct-FP8"`
`282`	`282`	`vllm_config.model_config = ModelConfig(`
`283`	`283`	`model=model_name, trust_remote_code=True, dtype=dtype, seed=42`
`284`	`284`	`)`