Ross Wightman
|
015fbe457a
|
Merge branch 'MengqingCao-npu_support' into device_amp_cleanup
|
2024-10-18 14:50:44 -07:00 |
|
Ross Wightman
|
81b59faf77
|
Merge branch 'npu_support' of github.com:MengqingCao/pytorch-image-models into MengqingCao-npu_support
|
2024-10-18 14:50:00 -07:00 |
|
Ross Wightman
|
1766a01f96
|
Cleanup some amp related behaviour to better support different (non-cuda) devices
|
2024-10-18 13:54:16 -07:00 |
MengqingCao
|
37c731ca37
|
fix device check
|
2024-10-17 12:38:02 +00:00 |
Ross Wightman
|
a852318b63
|
Merge pull request #2305 from NightMachinery/patch-2
mambaout.py: fixed bug
|
2024-10-16 14:39:43 -07:00 |
Feraidoon Mehri
|
ca20e102fe
|
mambaout.py: fixed bug
|
2024-10-17 01:03:28 +03:30 |
|
Ross Wightman
|
8cb2548962
|
Version 1.0.11
|
2024-10-16 14:14:44 -07:00 |
|
Ross Wightman
|
65e8e9ca12
|
Merge pull request #2304 from huggingface/intern300m
Add intern300m vit w/ converted timm weights. Fix #2300
|
2024-10-16 14:00:22 -07:00 |
|
Ross Wightman
|
89dffc5ff0
|
Another small fix for original mambaout models, no classifier nn.Linear when num_classe=0 on init
|
2024-10-16 12:36:36 -07:00 |
|
Ross Wightman
|
fad4538801
|
Elevate import deprecation warnings from DeprecationWarning to FutureWarning so messages are now seen
|
2024-10-16 11:30:01 -07:00 |
|
Ross Wightman
|
a1f379e712
|
Add intern300m vit w/ converted timm weights. Fix #2300
|
2024-10-16 10:29:06 -07:00 |
|
MengqingCao
|
234f975787
|
add npu support
|
2024-10-16 07:13:45 +00:00 |
|
Ross Wightman
|
60f517c883
|
Fix wrong name in _all_ for models._registry
|
2024-10-15 07:39:46 -07:00 |
|
Ross Wightman
|
b4a9a166c3
|
Version 1.0.10
|
2024-10-14 21:40:30 -07:00 |
|
Ross Wightman
|
c3052fa19e
|
Merge pull request #2298 from huggingface/preact_resnet18
Add resnet18/18d pre-act model configs for potential training.
|
2024-10-14 19:39:04 -07:00 |
|
Ross Wightman
|
abdf33145c
|
Add 34/34d pre-act resnet variants
|
2024-10-14 13:23:50 -07:00 |
|
Ross Wightman
|
c82ce86f8f
|
Add 384x384 mambaout_base_plus model weights
|
2024-10-14 12:28:57 -07:00 |
|
Ross Wightman
|
2703d155c8
|
Update README.md
|
2024-10-11 16:59:06 -07:00 |
|
Ross Wightman
|
9aaf579aa5
|
Update README.md
|
2024-10-11 15:14:01 -07:00 |
|
Ross Wightman
|
96cd3f6a7d
|
Update readme with latest
|
2024-10-11 15:10:05 -07:00 |
|
Ross Wightman
|
3236dff91d
|
Merge pull request #2266 from huggingface/mambaout
Mambaout
|
2024-10-11 12:16:48 -07:00 |
|
Ross Wightman
|
82ae247879
|
MambaOut weights on hub, configs finalized
|
2024-10-11 11:07:40 -07:00 |
|
Ross Wightman
|
7efb60c299
|
Add first_conv for mambaout
|
2024-10-09 14:11:40 -07:00 |
|
Ross Wightman
|
5dc5ee5b42
|
Add global_pool to mambaout __init__ and pass to heads
|
2024-10-09 14:11:40 -07:00 |
|
Ross Wightman
|
9d1dfe8dbe
|
Incorrectly named head_hidden_size
|
2024-10-09 14:11:40 -07:00 |
|
Ross Wightman
|
91e743f2dd
|
Mambaout tweaks
|
2024-10-09 14:11:40 -07:00 |
|
Ross Wightman
|
4542cf03f9
|
Add features_only, other bits to mambaout, define different base alternatives
|
2024-10-09 14:11:40 -07:00 |
|
Ross Wightman
|
c2da12c7e1
|
Update rw models, fix heads
|
2024-10-09 14:11:40 -07:00 |
|
Ross Wightman
|
f2086f51a0
|
Add mambaout builder support, pretrained weight remap
|
2024-10-09 14:11:40 -07:00 |
|
Ross Wightman
|
c6ef54eefa
|
Initial mambaout work
|
2024-10-09 14:11:40 -07:00 |
|
Ross Wightman
|
6ee638a095
|
Merge pull request #2299 from huggingface/siglip_update
Add i18n variant of so400m model w/ weights. Add two in1k fine-tunes
|
2024-10-09 10:25:32 -07:00 |
|
Ross Wightman
|
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 |
|
Ross Wightman
|
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 |
|
Ross Wightman
|
72f0edb7e8
|
missed first_conv for rnv2 18d
|
2024-10-08 12:38:54 -07:00 |
|
Ross Wightman
|
3ed603a2ce
|
Add resnet18/18d pre-act model configs for potential training. Fix #2289
|
2024-10-08 11:28:07 -07:00 |
|
Ross Wightman
|
41a79e0fcb
|
Add overlapped stem convnext zepto weights
|
2024-10-08 11:26:34 -07:00 |
|
Ross Wightman
|
4d4bdd64a9
|
Add --torchcompile-mode args to train, validation, inference, benchmark scripts
|
2024-10-02 15:17:53 -07:00 |
|
Ross Wightman
|
14d55a7cf3
|
Merge pull request #2292 from huggingface/inference_tests
Do full inference test against test vectors for test_* models
|
2024-10-02 11:48:21 -07:00 |
|
Ross Wightman
|
1a2d8bb5a0
|
Update pip install to use whl/cpu
|
2024-10-02 10:26:22 -07:00 |
|
Ross Wightman
|
036b1f0cc9
|
no 2.4.1 for cpu
|
2024-10-02 10:18:34 -07:00 |
|
Ross Wightman
|
95907e69c2
|
Further reduce atol for model comparison, move python 3.11 + torch 2.2 -> python 3.12 + torch 2.4.1
|
2024-10-02 10:05:27 -07:00 |
|
Ross Wightman
|
fde6719403
|
relax tolerance on inference test
|
2024-10-02 09:53:17 -07:00 |
|
Ross Wightman
|
0e27f302a0
|
Do full inference test against test vectors for test_* models
|
2024-10-02 09:39:26 -07:00 |
|
Ross Wightman
|
44f1a343e7
|
Merge pull request #2281 from huggingface/convnext_zepto
convnext zepto, rmsnorm experiments
|
2024-09-30 15:37:31 -07:00 |
|
Ross Wightman
|
545bd4056c
|
Tag along test_vit3 weights
|
2024-09-30 12:03:32 -07:00 |
|
Ross Wightman
|
69b687d4cc
|
Add zepto weights
|
2024-09-30 11:43:23 -07:00 |
|
Ross Wightman
|
c6e5557a5a
|
Mismatch pretrained_cfg
|
2024-09-30 11:43:23 -07:00 |
|
Ross Wightman
|
5d7bd2973e
|
convnext zepto, rmsnorm experiments
|
2024-09-30 11:43:23 -07:00 |
|
Ross Wightman
|
e3242a5258
|
Merge pull request #2277 from huggingface/more_tiny_test_models
Adding some more tiny test models to train...
|
2024-09-22 10:28:29 -07:00 |
|
Ross Wightman
|
305a360f08
|
Merge branch 'baorepo-patch-1'
|
2024-09-22 10:27:35 -07:00 |