2a84d68d02
Add some so150m vit w/ sbb recipe weights, and a ese_vovnet57b model with RA4 recipe
2025-01-18 15:51:57 -08:00
19aaea3c8f
Fix nn.Module type hints
2025-01-11 15:09:21 +01:00
2b251fb291
Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override
2025-01-06 11:28:39 -08:00
b0068ba5d0
Switch hf hub entries for new aimv2 / dfn weights to point to timm locations. Undo forced device for SDR linspace, part of another change.
2024-12-30 19:24:21 -08:00
1bf84b35c3
Update tests for aimv2 filtering
2024-12-30 19:24:21 -08:00
b33418713a
Add (almost) full set of aimv2 model instances. Switch back to unpacked SwiGLU. Verify correctness. Add DFN L/14 39B weight.
2024-12-30 19:24:21 -08:00
5804d92e4b
Switch aimv2 to used packed SwiGLU
2024-12-30 19:24:21 -08:00
a648a04834
Supporting aimv2 encoders
2024-12-30 19:24:21 -08:00
790decc89b
Add more pali(2) weights. Switch rest of models adapting open_clip weights to their own weight instances.
2024-12-27 14:00:41 -08:00
b12ecbd614
Move siglip timm weights to own repos
2024-12-27 14:00:41 -08:00
6fb7aaf37d
Switching to timm specific weight instances for open_clip image encoders to facilitate hf-hub: use in timm and new transformers TimmWrapper
2024-12-27 14:00:41 -08:00
3ae3f44288
Fix positional embedding resampling for non-square inputs in ViT
2024-11-06 16:03:27 -08:00
baa7242dd3
Fix c&p error, slight reformat
2024-10-23 21:51:09 -07:00
1b5cae681c
Update some clip pretrained weights to point to new hub locations, add a few missing weights
2024-10-23 21:51:09 -07:00
a1f379e712
Add intern300m vit w/ converted timm weights. Fix #2300
2024-10-16 10:29:06 -07:00
d9321b0e10
Add weights for fine-tuned siglip so400m. Add webli_i18n pretrained tags for the multi-lingual model variants (incl older base)
2024-10-09 09:04:44 -07:00
01b62264af
Add i18n variant of so400m model w/ weights. Add two in1k fine-tunes of original so400m 384x384 but at 378x378 (better matches patch14)
2024-10-08 23:40:24 -07:00
545bd4056c
Tag along test_vit3 weights
2024-09-30 12:03:32 -07:00
9067be6a30
Add weights for new tiny test models
2024-09-22 07:59:23 -07:00
a2f539f055
Add a few more test model defs in prep for weight upload
2024-09-21 11:38:38 -07:00
a50713ce6e
Fix #2272
2024-09-02 13:20:05 -07:00
9fcbf39cdc
Add remaining sbb vit betwixt/mediumd fine-tunes
2024-08-21 10:09:38 -07:00
2f3fed43b8
Fix hiera init with num_classes=0, fix weight tag names for sbb2 hiera/vit weights, add LayerScale/LayerScale2d to layers
2024-08-15 11:14:38 -07:00
fee91fdd41
Update Hiera model for abswin, more stable weight init, layer-scale. ImageNet-12k weights for hiera_small_abswin, and two of the sbb vits with improved reg4 init.
2024-08-14 12:22:40 -07:00
cec70b6779
Merge pull request #2225 from huggingface/small_things
...
Small things
2024-07-25 20:29:13 -07:00
d2240745d3
Fix issue where feature out_indices out of order after wrapping with FeatureGetterNet due to use of set()
2024-07-22 13:33:30 -07:00
9b2b8014e8
Add weights for test models
2024-07-22 10:08:57 -07:00
392b78aee7
set_input_size initial impl for vit & swin v1. Move HybridEmbed to own location in timm/layers
2024-07-17 15:25:48 -07:00
417cf7f871
Initialize weights of reg_token for ViT
2024-07-13 11:11:42 +08:00
d5afe106dc
Merge remote-tracking branch 'origin/tiny_test_models' into small_things
2024-07-09 12:49:57 -07:00
55101028bb
Rename test_tiny* -> test*. Fix ByobNet BasicBlock attn location and add test_byobnet model.
2024-07-09 11:53:11 -07:00
c33a001397
Fix typo
2024-06-24 21:54:38 +09:00
71101ebba0
Refactor vit pooling to add more reduction options, separately callable
2024-06-14 23:16:58 -07:00
9567cf6d84
Feature: add option global_pool='max' to VisionTransformer
...
Most of the CNNs have a max global pooling option. I would like to extend ViT to have this option.
2024-06-14 15:24:54 +02:00
66a0eb4673
Experimenting with tiny test models, how small can they go and be useful for regression tests?
2024-06-07 16:09:25 -07:00
a5a2ad2e48
Fix consistency, testing for forward_head w/ pre_logits, reset_classifier, models with pre_logits size != unpooled feature size
...
* add test that model supports forward_head(x, pre_logits=True)
* add head_hidden_size attr to all models and set differently from num_features attr when head has hidden layers
* test forward_features() feat dim == model.num_features and pre_logits feat dim == self.head_hidden_size
* more consistency in reset_classifier signature, add typing
* asserts in some heads where pooling cannot be disabled
Fix #2194
2024-06-07 13:53:00 -07:00
cc8a03daac
Add ConvStem and MobileCLIP hybrid model for B variant. Add full norm disable support to ConvNormAct layers
2024-06-06 09:15:27 -07:00
1b66ec7cf3
Fixup ViTamin, add hub weight reference
2024-06-03 17:14:03 -07:00
5dce710101
Add vit_little in12k + in12k-ft-in1k weights
2024-05-27 14:56:03 -07:00
84cb225ecb
Add in12k + 12k_ft_in1k vit_medium weights
2024-05-20 15:52:46 -07:00
83aee5c28c
Add explicit GAP (avg pool) variants of other SigLIP models.
2024-05-15 07:53:19 -07:00
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
a69863ad61
Merge pull request #2156 from huggingface/hiera
...
WIP Hiera implementation.
2024-05-13 14:58:12 -07:00
f7aa0a1a71
Add missing vit_wee weight
2024-05-13 12:05:47 -07:00
c838c4233f
Add typing to reset_classifier() on other models
2024-05-12 11:12:00 -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
cb57a96862
Fix early stop for efficientnet/mobilenetv3 fwd inter. Fix indices typing for all fwd inter.
2024-05-04 10:21:58 -07:00
f8979d4f50
Comment out time local files while testing new vit weights
2024-05-03 20:26:56 -07:00
Ross Wightman
Adam J. Stewart
Wojtek Jasiński
Ross Wightman
Promisery
Sejik
Fernando Cossio