Skip to content

Commit 363a65d

Browse files
committed
Add DINOv3 ConvNeXt weights.
1 parent 2d1bc8a commit 363a65d

File tree

2 files changed

+38
-2
lines changed

2 files changed

+38
-2
lines changed

timm/models/_hub.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -457,13 +457,18 @@ def push_to_hf_hub(
457457
)
458458

459459

460-
def generate_readme(model_card: dict, model_name: str):
461-
tags = model_card.get('tags', None) or ['image-classification', 'timm', 'transformers']
460+
def generate_readme(
461+
model_card: dict,
462+
model_name: str,
463+
task_name: str = 'image-classification',
464+
):
465+
tags = model_card.get('tags', None) or [task_name, 'timm', 'transformers']
462466
readme_text = "---\n"
463467
if tags:
464468
readme_text += "tags:\n"
465469
for t in tags:
466470
readme_text += f"- {t}\n"
471+
readme_text += f"pipeline_tag: {task_name}\n"
467472
readme_text += f"library_name: {model_card.get('library_name', 'timm')}\n"
468473
readme_text += f"license: {model_card.get('license', 'apache-2.0')}\n"
469474
if 'license_name' in model_card:

timm/models/convnext.py

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1061,6 +1061,37 @@ def _cfgv2(url='', **kwargs):
10611061
mean=OPENAI_CLIP_MEAN, std=OPENAI_CLIP_STD,
10621062
input_size=(3, 256, 256), pool_size=(8, 8), crop_pct=1.0, num_classes=1024),
10631063

1064+
# NOTE dinov3 convnext weights are under a specific license, and downstream outputs must be shared with this
1065+
# https://ai.meta.com/resources/models-and-libraries/dinov3-license/
1066+
'convnext_tiny.dinov3_lvd1689m': _cfg(
1067+
#hf_hub_id='timm/',
1068+
file='/data/x/dinov3_convnext_tiny_pretrain_lvd1689m-21b726bb.pth',
1069+
crop_pct=1.0,
1070+
num_classes=0,
1071+
license='dinov3',
1072+
),
1073+
'convnext_small.dinov3_lvd1689m': _cfg(
1074+
#hf_hub_id='timm/',
1075+
file='/data/x/dinov3_convnext_small_pretrain_lvd1689m-296db49d.pth',
1076+
crop_pct=1.0,
1077+
num_classes=0,
1078+
license='dinov3',
1079+
),
1080+
'convnext_base.dinov3_lvd1689m': _cfg(
1081+
#hf_hub_id='timm/',
1082+
file='/data/x/dinov3_convnext_base_pretrain_lvd1689m-801f2ba9.pth',
1083+
crop_pct=1.0,
1084+
num_classes=0,
1085+
license='dinov3',
1086+
),
1087+
'convnext_large.dinov3_lvd1689m': _cfg(
1088+
#hf_hub_id='timm/',
1089+
file='/data/x/dinov3_convnext_large_pretrain_lvd1689m-61fa432d.pth',
1090+
crop_pct=1.0,
1091+
num_classes=0,
1092+
license='dinov3',
1093+
),
1094+
10641095
"test_convnext.r160_in1k": _cfg(
10651096
hf_hub_id='timm/',
10661097
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5),

0 commit comments

Comments
 (0)