fix bug of VLLM_AVAILABLE not defined (#5497)

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