ruff fixes

HabanaAI · Sep 16, 2024 · e4fc78b · e4fc78b
1 parent 363de3c
commit e4fc78b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py b/vllm/model_executor/layers/quantization/compressed_tensors/compressed_tensors.py
@@ -233,7 +233,8 @@ def _get_scheme_from_parts(
         if is_activation_quantization_format(self.quant_format):
             if self._is_fp8_w8a8(weight_quant, input_quant):
                 is_fp8_w8a8_supported = self._check_scheme_supported(
-                    CompressedTensorsW8A8Fp8.get_min_capability(), error=False) \
+                    CompressedTensorsW8A8Fp8.get_min_capability(), 
+                    error=False) \
                     if torch.cuda.is_available() else True
                 if is_fp8_w8a8_supported:
                     return CompressedTensorsW8A8Fp8(