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.")