From a22fb1323ab71b89d86a2ebaa199d90dc4c26206 Mon Sep 17 00:00:00 2001 From: Ekaterina Aidova Date: Thu, 14 Nov 2024 17:56:11 +0400 Subject: [PATCH] Apply suggestions from code review Co-authored-by: Nikita Savelyev --- optimum/intel/openvino/modeling_visual_language.py | 1 - tests/openvino/test_modeling.py | 1 - 2 files changed, 2 deletions(-) diff --git a/optimum/intel/openvino/modeling_visual_language.py b/optimum/intel/openvino/modeling_visual_language.py index 73353d1d02..0baf37606d 100644 --- a/optimum/intel/openvino/modeling_visual_language.py +++ b/optimum/intel/openvino/modeling_visual_language.py @@ -454,7 +454,6 @@ def _from_pretrained( model_file_names[part] = f"openvino_{part}_model.xml" model_file_names[part + "_bin"] = f"openvino_{part}_model.bin" model_cls = MODEL_TYPE_TO_CLS_MAPPING[config.model_type] - quantization_config = model_cls._prepare_weight_quantization_config(quantization_config, load_in_8bit) compile_only = kwargs.get("compile_only", False) if os.path.isdir(model_id): # Load model from a local directory diff --git a/tests/openvino/test_modeling.py b/tests/openvino/test_modeling.py index 854cbd9bd7..ac7c480e57 100644 --- a/tests/openvino/test_modeling.py +++ b/tests/openvino/test_modeling.py @@ -246,7 +246,6 @@ def test_load_from_hub_and_save_visual_language_model(self): ).raw ) loaded_model = OVModelForVisualCausalLM.from_pretrained(model_id) - self.assertIsInstance(loaded_model.config, PretrainedConfig) self.assertIsInstance(loaded_model, MODEL_TYPE_TO_CLS_MAPPING[loaded_model.config.model_type]) self.assertIsInstance(loaded_model.vision_embeddings, OVVisionEmbedding) self.assertIsInstance(loaded_model.language_model, OVModelWithEmbedForCausalLM)