diff --git a/optimum/intel/openvino/modeling_timm.py b/optimum/intel/openvino/modeling_timm.py index 044e8bd3b6..2b20a6a746 100644 --- a/optimum/intel/openvino/modeling_timm.py +++ b/optimum/intel/openvino/modeling_timm.py @@ -163,7 +163,7 @@ def from_pretrained( pretrained_model_name_or_path: Union[str, os.PathLike], **kwargs, ): - timm_config_dict, _ = load_model_config_from_hf(pretrained_model_name_or_path) + timm_config_dict = load_model_config_from_hf(pretrained_model_name_or_path)[0] _, im_h, im_w = timm_config_dict.get("input_size", [3, 224, 224])