Skip to content

Commit a1794cc

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

File tree

2 files changed

+34
-2
lines changed

2 files changed

+34
-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: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1061,6 +1061,33 @@ 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+
crop_pct=1.0,
1069+
num_classes=0,
1070+
license='dinov3',
1071+
),
1072+
'convnext_small.dinov3_lvd1689m': _cfg(
1073+
hf_hub_id='timm/',
1074+
crop_pct=1.0,
1075+
num_classes=0,
1076+
license='dinov3',
1077+
),
1078+
'convnext_base.dinov3_lvd1689m': _cfg(
1079+
hf_hub_id='timm/',
1080+
crop_pct=1.0,
1081+
num_classes=0,
1082+
license='dinov3',
1083+
),
1084+
'convnext_large.dinov3_lvd1689m': _cfg(
1085+
hf_hub_id='timm/',
1086+
crop_pct=1.0,
1087+
num_classes=0,
1088+
license='dinov3',
1089+
),
1090+
10641091
"test_convnext.r160_in1k": _cfg(
10651092
hf_hub_id='timm/',
10661093
mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5),

0 commit comments

Comments
 (0)