Fix compatibility with timm latest release (#482)

huggingface · Nov 30, 2023 · bddeacd · bddeacd
1 parent eb687cf
commit bddeacd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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])