Apply suggestions from code review

Co-authored-by: Nikita Savelyev <[email protected]>
huggingface · Nov 14, 2024 · a22fb13 · a22fb13
1 parent 6391fa9
commit a22fb13
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git 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
@@ -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)