Update for MoE examples PR vllm-project#192

dbarbuzzi · Sep 23, 2024 · 5d27928 · 5d27928
1 parent 28abda4
commit 5d27928
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/examples/test_quantizing_moe.py b/tests/examples/test_quantizing_moe.py
@@ -55,7 +55,8 @@ def test_doc_example_command(self, example_dir: str, tmp_path: Path):
                 "deepseek_moe_w4a16.py",
                 marks=pytest.mark.skip(reason="exceptionally long run time"),
             ),
-            pytest.param("deepseek_moe_w8a8.py"),
+            pytest.param("deepseek_moe_w8a8_fp8.py"),
+            pytest.param("deepseek_moe_w8a8_int8.py"),
         ],
     )
     def test_deepseek_example_script(