Fixed issue

huggingface · Nov 13, 2023 · b8d98f4 · b8d98f4
1 parent 0cb7f19
commit b8d98f4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/optimum/intel/openvino/quantization.py b/optimum/intel/openvino/quantization.py
@@ -186,7 +186,7 @@ def quantize(
                 remove_unused_columns,
                 **kwargs,
             )
-        if isinstance(self.model, OVBaseDecoderModel) and self.model.use_cache:
+        elif isinstance(self.model, OVBaseDecoderModel) and self.model.use_cache:
             self._quantize_ovcausallm(
                 calibration_dataset,
                 save_directory,