7b3b11b63f
Support loading of paligemma weights into GAP variants of SigLIP ViT. Minor tweak to npz loading for packed transformer weights.
2024-05-14 15:44:37 -07:00
04462f554f
Fix typos in feat extract doc
2024-05-14 15:33:39 -07:00
df304ffbf2
the dataclass init needs to use the default factory pattern, according to Ross
2024-05-14 15:10:05 -04:00
cc5f2f6f70
version 1.0.2dev0
2024-05-13 15:25:15 -07:00
3bfd036b58
Add normalize flag to transforms factory, allow return of non-normalized native dtype torch.Tensors
2024-05-13 15:23:25 -07:00
a69863ad61
Merge pull request #2156 from huggingface/hiera
...
WIP Hiera implementation.
2024-05-13 14:58:12 -07:00
8848dad362
Update distributed.py
2024-05-13 16:55:42 -04:00
f7aa0a1a71
Add missing vit_wee weight
2024-05-13 12:05:47 -07:00
7a4e987b9f
Hiera weights on hub
2024-05-13 11:43:22 -07:00
23f09af08e
Merge branch 'main' into efficientnet_x
2024-05-12 21:31:08 -07:00
c838c4233f
Add typing to reset_classifier() on other models
2024-05-12 11:12:00 -07:00
3e03b2bf3f
Fix a few more hiera API issues
2024-05-12 11:11:45 -07:00
211d18d8ac
Move norm & pool into Hiera ClassifierHead. Misc fixes, update features_intermediate() naming
2024-05-11 23:37:35 -07:00
2ca45a4ff5
Merge remote-tracking branch 'upstream/main' into hiera
2024-05-11 15:43:05 -07:00
49de391470
Update feature_extraction.mdx
2024-05-11 12:14:02 -07:00
a193d20b7b
Update README.md
2024-05-11 11:01:05 -07:00
b2c10fec05
Merge pull request #2168 from huggingface/more_vit_better_getter_redux
...
A few more features_intermediate() models, AttentionExtract helper, related minor cleanup.
2024-05-11 08:13:50 -07:00
1d3ab176bc
Remove debug / staging code
2024-05-10 22:16:34 -07:00
aa4d06a11c
sbb vit weights on hub, testing
2024-05-10 17:15:01 -07:00
3582ca499e
Prepping weight push, benchmarking.
2024-05-10 14:14:06 -07:00
2bfa5e5d74
Remove JIT activations, take jit out of ME activations. Remove other instances of torch.jit.script. Breaks torch.compile and is much less performant. Remove SpaceToDepthModule
2024-05-06 16:32:49 -07:00
99d4c7d202
add ViTamin models
2024-05-05 02:50:14 -04:00
07535f408a
Add AttentionExtract helper module
2024-05-04 14:10:00 -07:00
45b7ae8029
forward_intermediates() support for byob/byoanet models
2024-05-04 14:06:52 -07:00
c4b8897e9e
attention -> attn in davit for model consistency
2024-05-04 14:06:11 -07:00
cb57a96862
Fix early stop for efficientnet/mobilenetv3 fwd inter. Fix indices typing for all fwd inter.
2024-05-04 10:21:58 -07:00
01dd01b70e
forward_intermediates() for MlpMixer models and RegNet.
2024-05-04 10:21:03 -07:00
f8979d4f50
Comment out time local files while testing new vit weights
2024-05-03 20:26:56 -07:00
3e7ab12af9
Merge pull request #2165 from huggingface/more_vits_better_getter
...
Add features_intermediate() support to coatnet, maxvit, swin*. Tweak feature extraction interfaces. Prep new vit weights while testing.
2024-05-03 19:26:01 -07:00
c719f7eb86
More forward_intermediates() updates
...
* add convnext, resnet, efficientformer, levit support
* remove kwargs only for fn so that torchscript isn't broken for all :(
* use reset_classifier() consistently in prune
2024-05-03 16:22:32 -07:00
301d0bb21f
Stricter check on pool_type for adaptive pooling module. Fix #2159
2024-05-03 16:16:51 -07:00
d6da4fb01e
Add forward_intermediates() to efficientnet / mobilenetv3 based models as an exercise.
2024-05-02 14:19:16 -07:00
c22efb9765
Add wee & little vits for some experiments
2024-05-02 10:51:35 -07:00
67332fce24
Add features_intermediate() support to coatnet, maxvit, swin* models. Refine feature interface. Start prep of new vit weights.
2024-04-30 16:56:33 -07:00
62516d525a
Merge pull request #2164 from ZFTurbo/main
...
Assert messages added
2024-04-30 10:13:38 -07:00
740f4983b3
Assert messages added
2024-04-30 10:10:02 +03:00
6de529bb3c
Merge pull request #2162 from huggingface/more_fwd_intermediates
...
Add forward_intermediates support for xcit, cait, and volo.
2024-04-29 21:09:57 -07:00
c6db4043cd
Update forward_intermediates for hiera to have its own fwd impl w/ early stopping. Remove return_intermediates bool from forward(). Still an fx issue with None mask arg :(
2024-04-29 17:23:37 -07:00
63970e4e61
Update tests for volo/cait/xcit forward_intermediates
2024-04-29 17:19:11 -07:00
9b9a356a04
Add forward_intermediates support for xcit, cait, and volo.
2024-04-29 16:30:45 -07:00
e8b08a4e7b
Disable jit test for Hiera, fair bit of work to fix.
2024-04-22 10:39:30 -07:00
ef147fd2fb
Add forward_intermediates API to Hiera for features_only=True support
2024-04-21 11:30:41 -07:00
d88bed6535
Bit more Hiera fiddling
2024-04-21 09:36:57 -07:00
8a54d2a930
WIP Hiera implementation. Fix #2083 . Trying to get image size adaptation to work.
2024-04-20 09:47:17 -07:00
e741370e2b
Update README.md
2024-04-11 10:16:39 -07:00
de15b8b828
Next release will be 1.0 :o
2024-04-11 08:55:27 -07:00
c8da47a773
Update version.py
2024-04-11 08:45:50 -07:00
d6b95520f1
Merge pull request #2136 from huggingface/vit_features_only
...
Exploring vit features_only via new forward_intermediates() API, inspired by #2131
2024-04-11 08:38:20 -07:00
24f6d4f7f8
Fix #2127 move to ema device
2024-04-10 21:29:09 -07:00
fe3cf542fa
Fix / improve tests for features
2024-04-10 21:14:02 -07:00
Ross Wightman
Beckschen
Ross Wightman
Setepenre
user-miner1