fix bug of VLLM_AVAILABLE not defined (#5497)

2025-04-18 15:59:03 +08:00
parent bed05878f6
commit e465b08ddb
1 changed files with 7 additions and 0 deletions
--- 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"]