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 |
|