diff --git a/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py b/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py index 71fc809f4d..2231b2ca9a 100644 --- a/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py +++ b/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py @@ -983,7 +983,7 @@ class CompressedTensorsKVCacheMethod(BaseKVCacheMethod): type_ = kv_cache_scheme.get("type") num_bits = kv_cache_scheme.get("num_bits") - if type_ != "float" and num_bits != 8: + if type_ != "float" or num_bits != 8: raise NotImplementedError( "Currently supported kv cache quantization is " "num_bits=8, type=float, however "