transformers
Add common test for `torch.export` and fix some vision models
#35124
Merged
Merged
Add common test for `torch.export` and fix some vision models #35124
qubvel
merged
46 commits into
huggingface:main
from
qubvel:test-torch-export
Add is_torch_greater_or_equal test decorator
645b6b27
Add common test for torch.export
98698b0d
Fix bit
208f9c48
Fix focalnet
b464e268
Fix imagegpt
a64f10bb
Fix seggpt
70fb0da6
Fix swin2sr
552a98ee
Enable torch.export test for vision models
5a2e0486
Enable test for video models
c63b95c5
Remove json
7dd1cce0
Enable for hiera
6e2e6014
Enable for ijepa
4d488198
Fix detr
b60030bc
Fic conditional_detr
6f7a0fdb
Fix maskformer
665f5833
Enable test maskformer
abb883cf
Fix test for deformable detr
3391cf0f
ydshieh
approved these changes
on 2024-12-18
Merge branch 'main' into test-torch-export
cd569af9
Fix custom kernels for export in rt-detr and deformable-detr
c22475b7
Enable test for all DPT
c1669169
Remove custom test for deformable detr
ed865778
Simplify test to use only kwargs for export
51683222
Add comment
50df345c
Move compile_compatible_method_lru_cache to utils
529e00a2
Fix beit export
79d3a8b4
Fix deformable detr
473a2010
Fix copies data2vec<->beit
1ada506d
Fix typos, update test to work with dict
cd5bc0fd
Add seed to the test
42600296
Enable test for vit_mae
90573519
Fix beit tests
239c1ce4
[run-slow] beit, bit, conditional_detr, data2vec, deformable_detr, de…
78ddd8f2
gmagogsfm
commented
on 2024-12-30
Merge branch 'main' into test-torch-export
43c2b481
Add vitpose test
db643ce3
Add textnet test
4a2062d5
Add dinov2 with registers
2fe4176e
Merge branch 'main' into test-torch-export
ca4b6acd
Update tests/test_modeling_common.py
1bc76e93
guangy10
approved these changes
on 2025-01-31
Merge branch 'main' into test-torch-export
2cc7e6a4
Switch to torch.testing.assert_close
b3c17ccb
Fix masformer
130bf0ab
Remove save-load from test
617403b7
Add dab_detr
b9b6aa6a
Add depth_pro
b1b7944e
Fix and test RT-DETRv2
8749a9a3
Fix dab_detr
c6e0f0f3
Reviewers
Assignees
No one assigned
Labels
Vision
run-slow
torch export
Milestone
No milestone
Login to write a write a comment.
Login via GitHub