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
|
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
|
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 |
|
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 |
|
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
|
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 |
|
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 |
|
Ryan
|
77788f4f92
|
add shvit
|
2025-04-29 06:24:44 +08:00 |
|
Ryan
|
b60a4fb329
|
add starnet
|
2025-04-29 05:38:42 +08:00 |
|
Ryan
|
081e6c2b96
|
update swiftformer
|
2025-04-29 05:19:53 +08:00 |
|
Ryan
|
7a87199bf8
|
add swiftformer
|
2025-04-29 00:58:10 +08:00 |
|
Xihua Dong
|
0c136f7dab
|
fix img_size type
|
2025-04-18 21:05:05 -07:00 |
|
Ross Wightman
|
3ff3899026
|
Add local-dir: schema support for model loading (config + weights) from folder
|
2025-04-17 12:32:19 -07:00 |
|
Ross Wightman
|
ceca5efdec
|
Remove torch_out from onnx export, no point without the export_ fn
|
2025-04-15 14:01:17 -07:00 |
|
Ross Wightman
|
0cae8a4cd8
|
Fix #2472, torch.onnx.export_ (with return output) finally removed :(
|
2025-04-15 14:01:17 -07:00 |
|
Ross Wightman
|
681be882e8
|
Fix arg merging of sknet, old seresnet. Fix #2470
|
2025-04-14 10:32:26 -07:00 |
|
Ross Wightman
|
98e9651952
|
Update version.py
Version 1.0.15, prep for a release
|
2025-02-22 10:50:21 -08:00 |
|
Adam J. Stewart
|
92682d8d4d
|
timm.models: explicitly export attributes
|
2025-02-21 14:19:39 -08:00 |
|