From 40f414ee24742b6d4269ad7f6ef8992e67a21508 Mon Sep 17 00:00:00 2001 From: Ella Charlaix Date: Wed, 20 Mar 2024 16:52:57 +0100 Subject: [PATCH] Remove warning --- optimum/intel/openvino/modeling_decoder.py | 21 --------------------- 1 file changed, 21 deletions(-) diff --git a/optimum/intel/openvino/modeling_decoder.py b/optimum/intel/openvino/modeling_decoder.py index 832c132615..10f0359a24 100644 --- a/optimum/intel/openvino/modeling_decoder.py +++ b/optimum/intel/openvino/modeling_decoder.py @@ -70,22 +70,6 @@ ``` """ -_SUPPORTED_ARCHITECTURES = { - "bart", - "blenderbot", - "blenderbot-small", - "bloom", - "codegen", - "gpt2", - "gpt-bigcode", - "gpt-neo", - "gpt-neox", - "llama", - "marian", - "opt", - "pegasus", -} - @add_start_docstrings( """ @@ -247,11 +231,6 @@ def _from_transformers( quantization_config: Optional[Union[OVWeightQuantizationConfig, Dict]] = None, **kwargs, ): - if config.model_type.replace("_", "-") not in _SUPPORTED_ARCHITECTURES: - logger.warning( - f"This architecture : {config.model_type} was not validated, only :{', '.join(_SUPPORTED_ARCHITECTURES)} architectures were " - "validated, use at your own risk." - ) save_dir = TemporaryDirectory() save_dir_path = Path(save_dir.name)