From 1a0cd69435cf6ddfef2b15e37cbddd94e13348f4 Mon Sep 17 00:00:00 2001 From: gebbissimo <4158176+cgebbe@users.noreply.github.com> Date: Tue, 19 Nov 2024 11:26:35 +0100 Subject: [PATCH] feat: allow to use hf-hub models for timm backbone (#34729) Currently a backbone name like 'hf-hub:bioptimus/H-optimus-0' throws an error, even though it could work. Co-authored-by: Christian Gebbe <> --- .../models/timm_backbone/modeling_timm_backbone.py | 5 ----- 1 file changed, 5 deletions(-) diff --git a/src/transformers/models/timm_backbone/modeling_timm_backbone.py b/src/transformers/models/timm_backbone/modeling_timm_backbone.py index ffe83daf7bc..ae25852d082 100644 --- a/src/transformers/models/timm_backbone/modeling_timm_backbone.py +++ b/src/transformers/models/timm_backbone/modeling_timm_backbone.py @@ -50,11 +50,6 @@ class TimmBackbone(PreTrainedModel, BackboneMixin): if config.backbone is None: raise ValueError("backbone is not set in the config. Please set it to a timm model name.") - # Certain timm models have the structure `model_name.version` e.g. vit_large_patch14_dinov2.lvd142m - base_backbone_model = config.backbone.split(".")[0] - if base_backbone_model not in timm.list_models(): - raise ValueError(f"backbone {base_backbone_model} is not supported by timm.") - if hasattr(config, "out_features") and config.out_features is not None: raise ValueError("out_features is not supported by TimmBackbone. Please use out_indices instead.")