2765 Commits

Author SHA1 Message Date
Ross Wightman
a22366e3ce
Merge pull request #2503 from huggingface/beit3_remap_clean
A cleaned up beit3 remap onto vision_transformer.py vit
2025-05-30 16:40:28 -07:00
Ross Wightman
211cf90721 Imports getting unwieldy in vision_transformer.py 2025-05-30 15:11:51 -07:00
Ross Wightman
1be7999993 Upload beit3 weights to hub, add pretrain weights 2025-05-30 14:36:41 -07:00
Ross Wightman
3a3d98bc38 Fix parallel blocks missing scale args and vitamin MLP 2025-05-29 11:34:19 -07:00
Ross Wightman
2ca94a6ce4 Compact _covert_beit3 fn 2025-05-29 10:52:39 -07:00
Ross Wightman
38c5f3bc63 A cleaned up beit3 remap onto vision_transformer.py vit 2025-05-29 10:33:09 -07:00
Ross Wightman
a0a30a6b48
Update README.md 2025-05-28 09:42:09 -07:00
Ross Wightman
3114a9f595
Merge pull request #2499 from brianhou0208/fast_model
Add SwiftFormer, SHViT, StarNet, FasterNet and GhostNetV3
2025-05-26 18:49:51 -07:00
Ross Wightman
4209788233 Updated faster models w/ hub weight locations, commented out some checkpoint filter fns and minor renames 2025-05-26 16:00:29 -07:00
Ryan
75823ab90e Merge branch 'main' into fast_model 2025-05-22 02:37:00 +08:00
Ross Wightman
e7925ea6d6 Add explicit disable of fc_norm to pe giant spatial variant 2025-05-21 07:31:36 -07:00
Ryan
bea1137583 Merge branch 'main' into fast_model 2025-05-17 22:25:34 +08:00
Ryan
236b00cf2d add ghostnetv3 2025-05-17 17:57:54 +08:00
Ross Wightman
edc37be1c3 Remove default 1000 classes from onnx export to use model default 2025-05-15 14:14:53 -07:00
Ross Wightman
f14f6507ab
Merge pull request #2487 from huggingface/eva_pe_integration
Add EVA ViT based PE (Perceptual Encoder) impl
2025-05-15 12:56:03 -07:00
Ross Wightman
cabd26d018 Fix missing device arg in train.py, fix #2488 2025-05-14 13:51:43 -07:00
Ross Wightman
6b302f27a3
Merge pull request #2480 from brianhou0208/tnt
Update TNT-(S/B) model weights and add feature extraction support
2025-05-14 12:27:21 -07:00
Ryan
1922ca5f1b fix fasternet 2025-05-15 02:28:17 +08:00
Ross Wightman
16d0b26e19 Fix torchscript issue with legacy tnt 2025-05-14 09:33:41 -07:00
Ross Wightman
74ad32a67e Updated tnt model weights on hub, add back legacy model in case bwd compat 2025-05-14 08:40:53 -07:00
Ryan
cb98094792 Merge branch 'main' into fast_model 2025-05-14 08:30:33 +08:00
Ryan
89d2952375 update group_matcher 2025-05-14 08:28:10 +08:00
Ryan
69b1fbcdc1 Merge branch 'main' into tnt 2025-05-11 22:45:48 +08:00
Ross Wightman
88b7ef6035 Disable dynamic_img_size default on PE models for now 2025-05-10 15:00:29 -07:00
Ross Wightman
fa7287faee Add EVA ViT based PE (Perceptual Encoder) impl 2025-05-10 13:13:12 -07:00
Ryan Ramos
81900a6bae remove download argument from torch_kwargs 2025-05-09 07:34:32 -07:00
atharva-pathak
59e59ff040 Update README.md
The previous link to Chris Hughes blog post was wrong, so updated the correct link
2025-05-08 08:43:18 -07:00
Ryan
d1140c1a0f fix pit & add to test 2025-05-08 07:17:19 -07:00
Ryan
2e9b2a76fb update some model 2025-05-08 07:17:19 -07:00
Ryan
99c25fa5c0 fix nest type error 2025-05-08 07:17:19 -07:00
Ryan
72b2a09106 fix final norm only apply at last indice 2025-05-08 07:17:19 -07:00
Ryan
e0ae4db8fc update FEAT_INTER_FILTERS list 2025-05-08 07:17:19 -07:00
Ryan
98d9cde8b4 support efficientformer_v2 2025-05-08 07:17:19 -07:00
Ryan
12def0d118 support efficientvit, edgenext, davit 2025-05-08 07:17:19 -07:00
Ryan
9aedecbb5f fix norm at last feat_idx 2025-05-08 07:17:19 -07:00
Ryan
e16d385592 support more models
inception_next, hgnet, gcvit, focalnet, inception_v4
2025-05-08 07:17:19 -07:00
Ryan
6b6beffa6b support mambaout, metaformer, nest, nextvit, pvt_v2 2025-05-08 07:17:19 -07:00
Ryan
2d3155908c fix tresnet and rdnet 2025-05-08 07:17:19 -07:00
Ryan
f8be741f0f support rexnet, resnetv2, repvit and repghostnet 2025-05-08 07:17:19 -07:00
Ryan
5e8cc616d4 support tiny_vit 2025-05-08 07:17:19 -07:00
Ryan
8befebd93c support vovnet 2025-05-08 07:17:19 -07:00
Ryan
411b892dbc support tresnet 2025-05-08 07:17:19 -07:00
Ross Wightman
fdcf7cf5c3 Check forward_intermediates features against forward_features output 2025-05-06 13:48:51 -07:00
Ryan
7fc0692843 update fasternet 2025-05-06 03:10:19 +08:00
Ryan
a3e66b14ea add fasternet 2025-05-06 02:59:33 +08:00
Ryan
91e6e1737e add test_models 2025-05-03 23:40:37 +08:00
Ryan
37bbac1256 Fix checkpoint_filter_fn 2025-05-03 15:41:18 +08:00
Ryan
fc0b6ad183 Fix default_cfgs 2025-05-03 08:40:07 +08:00
Ryan
848b8c3e57 Support features_only 2025-05-02 20:59:05 +08:00
Ryan
b37f0f7a76 Update tnt.py 2025-05-02 20:34:31 +08:00