Revert "Support nvidia/NVIDIA-Nemotron-Nano-9B-v2-FP8/NVFP4" (#12015)

2025-10-23 21:27:58 +08:00
parent 32852fe9e9
commit 6c18addb6f
10 changed files with 127 additions and 207 deletions
--- a/test/srt/layers/attention/mamba/test_mamba2_mixer.py
+++ b/test/srt/layers/attention/mamba/test_mamba2_mixer.py
@@ -1,6 +1,5 @@
 # Adapted from https://github.com/vllm-project/vllm/blob/2c58742dff8613a3bd7496f2008ce927e18d38d1/tests/kernels/mamba/test_mamba_mixer2.py

-
 from unittest.mock import patch

 import pytest
@@ -137,7 +136,3 @@ def mixer2_gated_norm_tensor_parallel(
        atol=5e-3,
        rtol=1e-3,
    )
-
-
-if __name__ == "__main__":
-    pytest.main([__file__])