fix bug of VLLM_AVAILABLE not defined (#5497)

e465b08d · liwenju0 · GitHub · bed05878 · e465b08d
Unverified Commit e465b08d authored Apr 18, 2025 by liwenju0 Committed by GitHub Apr 18, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

python/sglang/srt/layers/quantization/compressed_tensors/compressed_tensors.py ...ers/quantization/compressed_tensors/compressed_tensors.py +7 -0

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,6 +40,13 @@ from sglang.srt.layers.quantization.compressed_tensors.utils import (
    should_ignore_layer,
 )

+try:
+    import vllm
+
+    VLLM_AVAILABLE = True
+except ImportError:
+    VLLM_AVAILABLE = False
+
 logger = logging.getLogger(__name__)

 __all__ = ["CompressedTensorsLinearMethod"]