Add TimmWrapper #34564
qubvel
merged
92 commits into
huggingface:main
from
qubvel:timm-wrapper
Add files
39c42db9
Init
4b983757
Add TimmWrapperModel
aa494f16
Fix up
44d123e3
Some fixes
4b35ae25
Fix up
2b5db8f0
Remove old file
b07a5c98
Sort out import orders
e3a88b61
Fix some model loading
baffbe21
Compatible with pipeline and trainer
6fc50cf1
Fix up
ed00b41b
Delete test_timm_model_1/config.json
50b507ba
Remove accidentally commited files
de87e547
Delete src/transformers/models/modeling_timm_wrapper.py
a9070616
Remove empty imports; fix transformations applied
1f558416
Tidy up
23b38afd
Add image classifcation model to special cases
2f0aee1f
Create pretrained model; enable device_map='auto'
bff2e981
Enable most tests; fix init order
0c802533
Sort imports
a32de6a9
[run-slow] timm_wrapper
59f55d15
Pass num_classes into timm.create_model
666419f0
Remove train transforms from image processor
5540d32a
Update timm creation with pretrained=False
88f737c3
Fix gamma/beta issue for timm models
496d38da
Fixing gamma and beta renaming for timm models
4cfa51f9
Simplify config and model creation
3891975d
Remove attn_implementation diff
f878ab40
Fixup
b1f145f0
Docstrings
7d87a95c
Fix warning msg text according to test case
ae0425d0
Fix device_map auto
ec9eade9
Set dtype and device for pixel_values in forward
97670485
Enable output hidden states
d5478f6d
Enable tests for hidden_states and model parallel
b61048be
Remove default scriptable arg
72236b5c
Refactor inner model
e290f22b
Update timm version
0a08a1f9
rwightman
commented
on 2024-11-05
rwightman
commented
on 2024-11-05
Fix _find_mismatched_keys function
1faf9ece
Change inheritance for Classification model (fix weights loading with…
addbac8b
Minor bugfix
8ba45927
Disable save pretrained for image processor
2b993a70
Rename hook method for loaded keys correction
0b00cef1
Rename state dict keys on save, remove `timm_model` prefix, make chec…
485fe7aa
Managing num_labels <-> num_classes attributes
dede4e45
Enable loading checkpoints in Trainer to resume training
1ec132c8
copilot-pull-request-reviewer
commented
on 2024-11-19
Update error message for output_hidden_states
d96b257a
Add output hidden states test
6b1b6210
Decouple base and classification models
bb7d465b
Add more test cases
43902135
Add save-load-to-timm test
4233850a
Merge branch 'main' into timm-wrapper
ef96b6d0
Fix test name
c3bb39ab
Fixup
5444a6c7
rwightman
commented
on 2024-11-21
rwightman
commented
on 2024-11-21
Add do_pooling
06039d05
Add test for do_pooling
0b787359
Fix doc
5bb29502
Add tests for TimmWrapperModel
38b94232
Add validation for `num_classes=0` in timm config + test for DINO che…
c1cc1fa6
Adjust atol for test
3acafada
Fix docs
8e8f41b9
dev-ci
c5c288f1
dev-ci
a4ae76fb
Add tests for image processor
464874f0
Update docs
02fbd583
Update init to new format
eb9a66fd
Update docs in configuration
e08fe708
Fix some docs in image processor
77348045
Improve docs for modeling
051aceef
fix for is_timm_checkpoint
185fab8c
Update code examples
bca3279f
Fix header
672cc6d3
Fix typehint
983b9b20
Increase tolerance a bit
9c128cea
Merge branch 'main' into timm-wrapper
f2dba795
Fix Path
9bc887b0
Fixing model parallel tests
f92216f8
Disable "parallel" tests
42278a7b
Add comment for metadata
6b3ba3b2
Refactor AutoImageProcessor for timm wrapper loading
ff6efde8
Remove custom test_model_outputs_equivalence
a476610c
Add require_timm decorator
327095a8
Fix comment
7e0d2c6d
Make image processor work with older timm versions and tensor input
a10bc0de
Save config instead of whole model in image processor tests
a87fbf6a
Add docstring for `image_processor_filename`
90c1c88d
Sanitize kwargs for timm image processor
fd7b646f
Fix doc style
cdd38116
Update check for tensor input
3d1a76e9
Merge branch 'main' into timm-wrapper
cc0a3304
Update normalize
4edfe902
Remove _load_timm_model function
8b82e2b7
hongwhatamazon
commented
on 2025-04-28
Assignees
No one assigned
Labels
New model
Vision
run-slow
Milestone
No milestone
Login to write a write a comment.
Login via GitHub