fix compressed tensors WNA16 imports (#8142)

fd63b62e · Enrique Shockwave · GitHub · 719b29f2 · fd63b62e
Unverified Commit fd63b62e authored Jul 18, 2025 by Enrique Shockwave Committed by GitHub Jul 18, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

python/sglang/srt/layers/quantization/compressed_tensors/compressed_tensors.py ...ers/quantization/compressed_tensors/compressed_tensors.py +4 -1

No files found.
--- a/python/sglang/srt/layers/quantization/compressed_tensors/compressed_tensors.py
+++ b/python/sglang/srt/layers/quantization/compressed_tensors/compressed_tensors.py
@@ -40,7 +40,10 @@ from sglang.srt.layers.quantization.compressed_tensors.utils import (
 from sglang.srt.layers.quantization.unquant import UnquantizedLinearMethod

 try:
-    import vllm
+    from vllm.model_executor.layers.quantization.compressed_tensors.schemes.compressed_tensors_wNa16 import (
+        WNA16_SUPPORTED_BITS,
+        CompressedTensorsWNA16,
+    )

    VLLM_AVAILABLE = True
 except ImportError: