SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
8490d3159ce37087151e30211e33f59008a44233
HuggingFace_transformer/tests/models
History
NielsRogge 8490d3159c Add ViTPose (#30530) 
* First draft

* Make fixup

* Make forward pass worké

* Improve code

* More improvements

* More improvements

* Make predictions match

* More improvements

* Improve image processor

* Fix model tests

* Add classic decoder

* Convert classic decoder

* Verify image processor

* Fix classic decoder logits

* Clean up

* Add post_process_pose_estimation

* Improve post_process_pose_estimation

* Use AutoBackbone

* Add support for MoE models

* Fix tests, improve num_experts%

* Improve variable names

* Make fixup

* More improvements

* Improve post_process_pose_estimation

* Compute centers and scales

* Improve postprocessing

* More improvements

* Fix ViTPoseBackbone tests

* Add docstrings, fix image processor tests

* Update index

* Use is_cv2_available

* Add model to toctree

* Add cv2 to doc tests

* Remove script

* Improve conversion script

* Add coco_to_pascal_voc

* Add box_to_center_and_scale to image_transforms

* Update tests

* Add integration test

* Fix merge

* Address comments

* Replace numpy by pytorch, improve docstrings

* Remove get_input_embeddings

* Address comments

* Move coco_to_pascal_voc

* Address comment

* Fix style

* Address comments

* Fix test

* Address comment

* Remove udp

* Remove comment

* [WIP] need to check if the numpy function is same as cv

* add scipy affine_transform

* Update src/transformers/models/vitpose/image_processing_vitpose.py

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* refactor convert

* add output_shape

* add atol 5e-2

* Use hf_hub_download in conversion script

* make box_to_center more applicable

* skipt test_get_set_embedding

* fix to accept array and fix CI

* add co-contributor

* make it to tensor type output

* add torch

* change to torch tensor

* add more test

* minor change

* CI test change

* import torch should be above ImageProcessor

* make style

* try not use torch in def

* Update src/transformers/models/vitpose/image_processing_vitpose.py

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* Update src/transformers/models/vitpose_backbone/configuration_vitpose_backbone.py

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* Update src/transformers/models/vitpose_backbone/modeling_vitpose_backbone.py

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* Update src/transformers/models/vitpose/modeling_vitpose.py

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* fix

* fix

* add caution

* make more detail about dataset_index

* Update src/transformers/models/vitpose/modeling_vitpose.py

Co-authored-by: Sangbum Daniel Choi <34004152+SangbumChoi@users.noreply.github.com>

* Update src/transformers/models/vitpose/image_processing_vitpose.py

Co-authored-by: Sangbum Daniel Choi <34004152+SangbumChoi@users.noreply.github.com>

* add docs

* Update docs/source/en/model_doc/vitpose.md

* Update src/transformers/models/vitpose/configuration_vitpose.py

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* Update src/transformers/__init__.py

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* Revert "Update src/transformers/__init__.py"

This reverts commit 7ffa504450bb9dbccf9c7ea668441b98a1939d5c.

* change name

* Update src/transformers/models/vitpose/image_processing_vitpose.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Update tests/models/vitpose/test_modeling_vitpose.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Update docs/source/en/model_doc/vitpose.md

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Update src/transformers/models/vitpose/modeling_vitpose.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Update src/transformers/models/vitpose_backbone/modeling_vitpose_backbone.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Update src/transformers/models/vitpose/image_processing_vitpose.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* move vitpose only function to image_processor

* raise valueerror when using timm backbone

* use out_indices

* Update src/transformers/models/vitpose/image_processing_vitpose.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* remove camel-case of def flip_back

* rename vitposeEstimatorOutput

* Update src/transformers/models/vitpose_backbone/modeling_vitpose_backbone.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* fix confused camelcase of MLP

* remove in-place logic

* clear scale description

* make consistent batch format

* docs update

* formatting docstring

* add batch tests

* test docs change

* Update src/transformers/models/vitpose/image_processing_vitpose.py

* Update src/transformers/models/vitpose/configuration_vitpose.py

* chagne ViT to Vit

* change to enable MoE

* make fix-copies

* Update docs/source/en/model_doc/vitpose.md

Co-authored-by: NielsRogge <48327001+NielsRogge@users.noreply.github.com>

* extract udp

* add more described docs

* simple fix

* change to accept target_size

* make style

* Update src/transformers/models/vitpose/image_processing_vitpose.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* Update src/transformers/models/vitpose/configuration_vitpose.py

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* change to `verify_backbone_config_arguments`

* Update docs/source/en/model_doc/vitpose.md

Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>

* remove unnecessary copy

* make config immutable

* enable gradient checkpointing

* update inappropriate docstring

* linting docs

* split function for visibility

* make style

* check isinstances

* change to acceptable use_pretrained_backbone

* make style

* remove copy in docs

* Update src/transformers/models/vitpose_backbone/modeling_vitpose_backbone.py

Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>

* Update docs/source/en/model_doc/vitpose.md

Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>

* Update src/transformers/models/vitpose/modeling_vitpose.py

Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>

* simple fix + make style

* change input config of activation function to string

* Update docs/source/en/model_doc/vitpose.md

Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>

* tmp docs

* delete index.md

* make fix-copies

* simple fix

* change conversion to sam2/mllama style

* Update src/transformers/models/vitpose/image_processing_vitpose.py

Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>

* Update src/transformers/models/vitpose/image_processing_vitpose.py

Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>

* refactor convert

* add supervision

* Update src/transformers/models/vitpose_backbone/modeling_vitpose_backbone.py

Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>

* remove reduntant def

* seperate code block for visualization

* add validation for num_moe

* final commit

* add labels

* [run-slow] vitpose, vitpose_backbone

* Update src/transformers/models/vitpose/convert_vitpose_to_hf.py

Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>

* enable all conversion

* final commit

* [run-slow] vitpose, vitpose_backbone

* ruff check --fix

* [run-slow] vitpose, vitpose_backbone

* rename split module

* [run-slow] vitpose, vitpose_backbone

* fix pos_embed

* Simplify init

* Revert "fix pos_embed"

This reverts commit 2c56a4806e30bc9b5753b142fa04b913306c54ff.

* refactor single loop

* allow flag to enable custom model

* efficiency of MoE to not use unused experts

* make style

* Fix range -> arange to avoid warning

* Revert MOE router, a new one does not work

* Fix postprocessing a bit (labels)

* Fix type hint

* Fix docs snippets

* Fix links to checkpoints

* Fix checkpoints in tests

* Fix test

* Add image to docs

---------

Co-authored-by: Niels Rogge <nielsrogge@nielss-mbp.home>
Co-authored-by: Niels Rogge <nielsrogge@Nielss-MacBook-Pro.local>
Co-authored-by: sangbumchoi <danielsejong55@gmail.com>
Co-authored-by: Sangbum Daniel Choi <34004152+SangbumChoi@users.noreply.github.com>
Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
Co-authored-by: Pavel Iakubovskii <qubvel@gmail.com>
2025-01-08 16:02:14 +00:00
..
albert
Remove @slow for test_eager_matches_sdpa_inference (#34558)
2024-11-05 16:10:42 +01:00
align
Load sub-configs from composite configs (#34410)
2024-11-05 11:34:01 +01:00
altclip
Load sub-configs from composite configs (#34410)
2024-11-05 11:34:01 +01:00
aria
Aurevoir PyTorch 1 (#35358)
2024-12-20 14:36:31 +01:00
audio_spectrogram_transformer
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
auto
Fall back to slow image processor in ImageProcessingAuto when no fast processor available (#34785)
2024-12-15 14:00:36 -05:00
autoformer
Rename test_model_common_attributes -> test_model_get_set_embeddings (#31321)
2024-06-07 19:40:26 +01:00
bamba
Add the Bamba Model (#34982)
2024-12-18 20:18:17 +01:00
bark
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
bart
Tests: move generate tests to the right mixin and delete redundant tests (#34464)
2024-10-30 10:59:08 +00:00
barthez
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
bartpho
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
beit
Add sdpa for Beit (#34941)
2024-12-17 14:44:47 +01:00
bert
Tests: move generate tests to the right mixin and delete redundant tests (#34464)
2024-10-30 10:59:08 +00:00
bert_generation
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
bert_japanese
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
bertweet
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
big_bird
[pytes collection] Fix flax test collection (#34004)
2024-10-07 18:11:13 +02:00
bigbird_pegasus
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
biogpt
Generate: unify LogitsWarper and LogitsProcessor (#32626)
2024-08-16 11:20:41 +01:00
bit
Use self.config_tester.run_common_tests() (#31431)
2024-06-19 10:18:08 +01:00
blenderbot
Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659)
2024-08-23 11:12:53 +01:00
blenderbot_small
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
blip
Load sub-configs from composite configs (#34410)
2024-11-05 11:34:01 +01:00
blip_2
BLIP: fix generation after hub update (#34876)
2024-11-25 10:41:55 +01:00
bloom
VLMs: enable generation tests (#33533)
2024-09-19 12:04:24 +02:00
bridgetower
rename all test_processing_*.py to test_processor_*.py (#33878)
2024-10-02 16:43:43 +02:00
bros
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
byt5
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
camembert
VLM: special multimodal Tokenizer (#34461)
2024-11-04 16:37:51 +01:00
canine
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
chameleon
Add image text to text pipeline (#34170)
2024-10-31 15:48:11 -04:00
chinese_clip
adding positional encoder changes and tests (#32600)
2024-09-25 19:05:01 +01:00
clap
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
clip
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
clipseg
Load sub-configs from composite configs (#34410)
2024-11-05 11:34:01 +01:00
clvp
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
code_llama
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
codegen
Generation tests: update imagegpt input name, remove unused functions (#33663)
2024-09-24 16:40:48 +01:00
cohere
Add Cohere2 model (#35224)
2024-12-13 09:35:50 +01:00
cohere2
Add Cohere2 model (#35224)
2024-12-13 09:35:50 +01:00
colpali
Add ColPali to 🤗 transformers (#33736)
2024-12-17 11:26:43 +01:00
conditional_detr
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
convbert
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
convnext
Update kwargs validation for preprocess with decorator (#32024)
2024-08-06 11:33:05 +01:00
convnextv2
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
cpm
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
cpmant
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
ctrl
avoid calling gc.collect and cuda.empty_cache (#34514)
2024-10-31 16:36:13 +01:00
cvt
Use self.config_tester.run_common_tests() (#31431)
2024-06-19 10:18:08 +01:00
dac
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
data2vec
Add sdpa for Beit (#34941)
2024-12-17 14:44:47 +01:00
dbrx
Generate: end-to-end compilation (#30788)
2024-07-29 10:52:13 +01:00
deberta
[Deberta/Deberta-v2] Refactor code base to support compile, export, and fix LLM (#22105)
2024-11-25 10:43:16 +01:00
deberta_v2
[Deberta/Deberta-v2] Refactor code base to support compile, export, and fix LLM (#22105)
2024-11-25 10:43:16 +01:00
decision_transformer
Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659)
2024-08-23 11:12:53 +01:00
deformable_detr
Add Image Processor Fast Deformable DETR (#34353)
2024-11-19 11:18:58 -05:00
deit
Update kwargs validation for preprocess with decorator (#32024)
2024-08-06 11:33:05 +01:00
depth_anything
fix(DPT,Depth-Anything) torch.export (#34103)
2024-11-20 11:31:21 +01:00
detr
Fall back to slow image processor in ImageProcessingAuto when no fast processor available (#34785)
2024-12-15 14:00:36 -05:00
diffllama
Add diffllama (#34083)
2025-01-07 11:34:56 +01:00
dinat
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
dinov2
Sdpa dino v2 (#33403)
2024-09-21 01:58:00 +01:00
dinov2_with_registers
Add DINOv2 with registers (#35348)
2024-12-24 13:21:59 +01:00
distilbert
DistilBERT is ExecuTorch compatible (#34475)
2024-11-05 13:41:48 +01:00
dit
Update old existing feature extractor references (#24552)
2023-06-29 10:17:36 +01:00
donut
rename all test_processing_*.py to test_processor_*.py (#33878)
2024-10-02 16:43:43 +02:00
dpr
Remove static pretrained maps from the library's internals (#29112)
2024-03-25 10:33:38 +01:00
dpt
fix(DPT,Depth-Anything) torch.export (#34103)
2024-11-20 11:31:21 +01:00
efficientnet
CI: fix efficientnet pipeline timeout and prevent future similar issues due to large image size (#33123)
2024-08-27 11:58:27 +01:00
electra
[pytes collection] Fix flax test collection (#34004)
2024-10-07 18:11:13 +02:00
encodec
Fix bug when requesting input normalization with EnCodec (#34756)
2025-01-07 11:50:02 +01:00
encoder_decoder
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
ernie
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
esm
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
falcon
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
falcon_mamba
Aurevoir PyTorch 1 (#35358)
2024-12-20 14:36:31 +01:00
fastspeech2_conformer
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
flaubert
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
flava
Load sub-configs from composite configs (#34410)
2024-11-05 11:34:01 +01:00
fnet
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
focalnet
Use self.config_tester.run_common_tests() (#31431)
2024-06-19 10:18:08 +01:00
fsmt
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
funnel
update ruff version (#30932)
2024-05-22 06:40:15 +02:00
fuyu
VLMs: enable generation tests - last batch (#34484)
2024-11-21 11:00:22 +01:00
gemma
Tests: move generate tests to the right mixin and delete redundant tests (#34464)
2024-10-30 10:59:08 +00:00
gemma2
[FlexAttention] Update gemma2 (#34942)
2024-11-27 11:50:48 +01:00
git
Add image text to text pipeline (#34170)
2024-10-31 15:48:11 -04:00
glm
Remove @slow for test_eager_matches_sdpa_inference (#34558)
2024-11-05 16:10:42 +01:00
glpn
Add post_process_depth_estimation for GLPN (#34413)
2024-10-28 19:44:20 +01:00
gpt2
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
gpt_bigcode
Aurevoir PyTorch 1 (#35358)
2024-12-20 14:36:31 +01:00
gpt_neo
Generation tests: update imagegpt input name, remove unused functions (#33663)
2024-09-24 16:40:48 +01:00
gpt_neox
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
gpt_neox_japanese
Compile compatibilty for decoder-only models (#32617)
2024-09-09 10:59:04 +02:00
gpt_sw3
🚨 No more default chat templates (#31733)
2024-07-24 17:36:32 +01:00
gptj
Aurevoir PyTorch 1 (#35358)
2024-12-20 14:36:31 +01:00
granite
Fix some fa2 tests (#35340)
2024-12-19 17:05:25 +01:00
granitemoe
Fix some fa2 tests (#35340)
2024-12-19 17:05:25 +01:00
grounding_dino
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
groupvit
Load sub-configs from composite configs (#34410)
2024-11-05 11:34:01 +01:00
herbert
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
hiera
🚨 Fix torch.jit.trace for interpolate_pos_encoding in all vision models (#33226)
2024-09-05 16:17:34 +02:00
hubert
Support BatchNorm in Hubert pos_conv_emb as in fairseq (#34389)
2024-12-10 14:18:23 +01:00
ibert
[warnings] fix E721 warnings (#32223)
2024-07-25 15:12:23 +02:00
idefics
Aurevoir PyTorch 1 (#35358)
2024-12-20 14:36:31 +01:00
idefics2
Aurevoir PyTorch 1 (#35358)
2024-12-20 14:36:31 +01:00
idefics3
bugfix Idefics3 processor - handle gracefully cases with text and no images (#35363)
2024-12-23 16:59:01 +01:00
ijepa
Update I-JEPA checkpoints path (#35120)
2024-12-06 13:42:51 +00:00
imagegpt
Simplify running tests in a subprocess (#34213)
2024-10-29 10:48:57 +01:00
informer
tests: fix pytorch tensor placement errors (#33485)
2024-09-25 12:21:53 +01:00
instructblip
BLIP: fix generation after hub update (#34876)
2024-11-25 10:41:55 +01:00
instructblipvideo
Load sub-configs from composite configs (#34410)
2024-11-05 11:34:01 +01:00
jamba
Tests: move generate tests to the right mixin and delete redundant tests (#34464)
2024-10-30 10:59:08 +00:00
jetmoe
Tests: move generate tests to the right mixin and delete redundant tests (#34464)
2024-10-30 10:59:08 +00:00
kosmos2
VLMs: enable generation tests - last batch (#34484)
2024-11-21 11:00:22 +01:00
layoutlm
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
layoutlmv2
Fix #34494 assistant tokens when truncated (#34531)
2024-11-05 15:10:15 +00:00
layoutlmv3
Fix #34494 assistant tokens when truncated (#34531)
2024-11-05 15:10:15 +00:00
layoutxlm
Fix #34494 assistant tokens when truncated (#34531)
2024-11-05 15:10:15 +00:00
led
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
levit
Update kwargs validation for preprocess with decorator (#32024)
2024-08-06 11:33:05 +01:00
lilt
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
llama
Fix some fa2 tests (#35340)
2024-12-19 17:05:25 +01:00
llava
VLMs: major clean up 🧼 (#34502)
2025-01-08 10:35:23 +01:00
llava_next
VLMs: major clean up 🧼 (#34502)
2025-01-08 10:35:23 +01:00
llava_next_video
VLMs: major clean up 🧼 (#34502)
2025-01-08 10:35:23 +01:00
llava_onevision
Aurevoir PyTorch 1 (#35358)
2024-12-20 14:36:31 +01:00
longformer
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
longt5
T5 compile compatibilty (#34089)
2024-10-22 08:23:53 +02:00
luke
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
lxmert
Refactor: Removed un-necessary object base class (#32230)
2024-07-26 10:33:02 +02:00
m2m_100
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
mamba
Aurevoir PyTorch 1 (#35358)
2024-12-20 14:36:31 +01:00
mamba2
Aurevoir PyTorch 1 (#35358)
2024-12-20 14:36:31 +01:00
marian
Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659)
2024-08-23 11:12:53 +01:00
markuplm
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
mask2former
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
maskformer
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
mbart
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
mbart50
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
megatron_bert
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
megatron_gpt2
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
mgp_str
Remove repeated prepare_images in processor tests (#33163)
2024-09-09 13:20:27 +01:00
mimi
Fix test_eager_matches_sdpa_inference for XPU backend (#34889)
2024-12-02 16:21:04 +01:00
mistral
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
mixtral
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
mllama
Offloaded cache: fix generate (#34921)
2024-11-28 15:05:56 +01:00
mluke
fix: Updated test_embeded_special_tokens for luke and mluke models (#32413)
2024-08-05 15:19:42 +01:00
mobilebert
MobileBERT is ExecuTorch compatible (#34473)
2024-10-29 16:14:31 +01:00
mobilenet_v1
Update kwargs validation for preprocess with decorator (#32024)
2024-08-06 11:33:05 +01:00
mobilenet_v2
Update kwargs validation for preprocess with decorator (#32024)
2024-08-06 11:33:05 +01:00
mobilevit
Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659)
2024-08-23 11:12:53 +01:00
mobilevitv2
Rename test_model_common_attributes -> test_model_get_set_embeddings (#31321)
2024-06-07 19:40:26 +01:00
modernbert
Add ModernBERT to Transformers (#35158)
2024-12-19 14:03:35 +01:00
moshi
VLM: special multimodal Tokenizer (#34461)
2024-11-04 16:37:51 +01:00
mpnet
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
mpt
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
mra
Fix Failed tests with mobile bert resize tokens embedding (#33950)
2024-10-09 11:23:50 +01:00
mt5
Tests: move generate tests to the right mixin and delete redundant tests (#34464)
2024-10-30 10:59:08 +00:00
musicgen
Fix test_eager_matches_sdpa_inference for XPU backend (#34889)
2024-12-02 16:21:04 +01:00
musicgen_melody
Make test_generate_with_static_cache even less flaky (#34995)
2024-12-20 16:03:26 +01:00
mvp
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
myt5
[Red CIs] Fix hub failures (#34001)
2024-10-07 10:56:24 +02:00
nemotron
Remove @slow for test_eager_matches_sdpa_inference (#34558)
2024-11-05 16:10:42 +01:00
nllb
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
nllb_moe
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
nougat
Update kwargs validation for preprocess with decorator (#32024)
2024-08-06 11:33:05 +01:00
nystromformer
update ruff version (#30932)
2024-05-22 06:40:15 +02:00
olmo
Tests: upgrade test_eager_matches_sdpa_generate (#34386)
2024-10-25 11:55:07 +01:00
olmo2
Rename OLMo November to OLMo2 (#34864)
2024-11-25 16:31:22 +01:00
olmoe
Tests: upgrade test_eager_matches_sdpa_generate (#34386)
2024-10-25 11:55:07 +01:00
omdet_turbo
Add support for custom inputs and batched inputs in ProcessorTesterMixin (#33711)
2024-10-01 23:52:03 +02:00
oneformer
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
openai
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
opt
Tests: upgrade test_eager_matches_sdpa_generate (#34386)
2024-10-25 11:55:07 +01:00
owlv2
owlvit/2 dynamic input resolution (#34764)
2024-12-21 08:51:09 +00:00
owlvit
owlvit/2 dynamic input resolution (#34764)
2024-12-21 08:51:09 +00:00
paligemma
Aurevoir PyTorch 1 (#35358)
2024-12-20 14:36:31 +01:00
patchtsmixer
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
patchtst
Rename test_model_common_attributes -> test_model_get_set_embeddings (#31321)
2024-06-07 19:40:26 +01:00
pegasus
Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659)
2024-08-23 11:12:53 +01:00
pegasus_x
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
perceiver
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
persimmon
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
phi
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
phi3
Phi3: fix attn for sliding window (#33586)
2024-10-10 11:50:39 +02:00
phimoe
fix red check-copies (#33964)
2024-10-04 22:45:37 +02:00
phobert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
pix2struct
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
pixtral
Add compile test for fast image processor (#35184)
2024-12-23 13:12:45 -05:00
plbart
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
poolformer
Update kwargs validation for preprocess with decorator (#32024)
2024-08-06 11:33:05 +01:00
pop2piano
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
prophetnet
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
pvt
Update kwargs validation for preprocess with decorator (#32024)
2024-08-06 11:33:05 +01:00
pvt_v2
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
qwen2
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
qwen2_audio
[Qwen2Audio] handle input ids expansion during processing (#35534)
2025-01-07 16:47:27 +01:00
qwen2_moe
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
qwen2_vl
Fix Qwen2VL processor to handle odd number of frames (#35431)
2025-01-08 13:49:00 +01:00
rag
[tests] make cuda-only tests device-agnostic (#35222)
2024-12-18 10:14:22 +01:00
recurrent_gemma
Tests: move generate tests to the right mixin and delete redundant tests (#34464)
2024-10-30 10:59:08 +00:00
reformer
Generation tests: don't rely on main input name (#34228)
2024-10-21 10:00:14 +02:00
regnet
[pytes collection] Fix flax test collection (#34004)
2024-10-07 18:11:13 +02:00
rembert
VLM: special multimodal Tokenizer (#34461)
2024-11-04 16:37:51 +01:00
resnet
[pytes collection] Fix flax test collection (#34004)
2024-10-07 18:11:13 +02:00
roberta
Roberta is ExecuTorch compatible (#34425)
2024-10-30 08:36:45 +00:00
roberta_prelayernorm
[pytes collection] Fix flax test collection (#34004)
2024-10-07 18:11:13 +02:00
roc_bert
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
roformer
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
rt_detr
Fall back to slow image processor in ImageProcessingAuto when no fast processor available (#34785)
2024-12-15 14:00:36 -05:00
rwkv
Aurevoir PyTorch 1 (#35358)
2024-12-20 14:36:31 +01:00
sam
uniformize kwargs for SAM (#34578)
2024-12-23 13:54:57 +01:00
seamless_m4t
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
seamless_m4t_v2
Make test_generate_with_static_cache even less flaky (#34995)
2024-12-20 16:03:26 +01:00
segformer
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
seggpt
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
sew
Fix parametrization-based weight norm (#33275)
2024-09-17 08:05:21 +02:00
sew_d
Fix parametrization-based weight norm (#33275)
2024-09-17 08:05:21 +02:00
siglip
Load sub-configs from composite configs (#34410)
2024-11-05 11:34:01 +01:00
speech_encoder_decoder
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
speech_to_text
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
speecht5
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
splinter
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
squeezebert
Refactor: Removed un-necessary object base class (#32230)
2024-07-26 10:33:02 +02:00
stablelm
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
starcoder2
Tests: move generate tests to the right mixin and delete redundant tests (#34464)
2024-10-30 10:59:08 +00:00
superpoint
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
swiftformer
Rename test_model_common_attributes -> test_model_get_set_embeddings (#31321)
2024-06-07 19:40:26 +01:00
swin
Use self.config_tester.run_common_tests() (#31431)
2024-06-19 10:18:08 +01:00
swin2sr
Update kwargs validation for preprocess with decorator (#32024)
2024-08-06 11:33:05 +01:00
swinv2
Fix dtype casting in swinv2 and swinv2sr to allow non-FP32 inference (#31589)
2024-06-26 18:46:48 +01:00
switch_transformers
T5 compile compatibilty (#34089)
2024-10-22 08:23:53 +02:00
t5
Tests: move generate tests to the right mixin and delete redundant tests (#34464)
2024-10-30 10:59:08 +00:00
table_transformer
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
tapas
Aurevoir PyTorch 1 (#35358)
2024-12-20 14:36:31 +01:00
textnet
Add TextNet (#34979)
2025-01-08 09:52:51 +01:00
time_series_transformer
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
timesformer
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
timm_backbone
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
timm_wrapper
Enable auto task for timm models in pipeline (#35531)
2025-01-08 15:14:17 +00:00
trocr
🚨🚨🚨 Uniformize kwargs for TrOCR Processor (#34587)
2024-11-29 11:58:11 +00:00
tvp
Update kwargs validation for preprocess with decorator (#32024)
2024-08-06 11:33:05 +01:00
udop
Fix #34494 assistant tokens when truncated (#34531)
2024-11-05 15:10:15 +00:00
umt5
Tests: move generate tests to the right mixin and delete redundant tests (#34464)
2024-10-30 10:59:08 +00:00
unispeech
Remove trust_remote_code when loading Libri Dummy (#31748)
2024-07-23 14:54:38 +08:00
unispeech_sat
Remove trust_remote_code when loading Libri Dummy (#31748)
2024-07-23 14:54:38 +08:00
univnet
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
upernet
Use self.config_tester.run_common_tests() (#31431)
2024-06-19 10:18:08 +01:00
video_llava
VLMs: major clean up 🧼 (#34502)
2025-01-08 10:35:23 +01:00
videomae
Remove @slow for test_eager_matches_sdpa_inference (#34558)
2024-11-05 16:10:42 +01:00
vilt
Update kwargs validation for preprocess with decorator (#32024)
2024-08-06 11:33:05 +01:00
vipllava
VLMs: major clean up 🧼 (#34502)
2025-01-08 10:35:23 +01:00
vision_encoder_decoder
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
vision_text_dual_encoder
Fall back to slow image processor in ImageProcessingAuto when no fast processor available (#34785)
2024-12-15 14:00:36 -05:00
visual_bert
update ruff version (#30932)
2024-05-22 06:40:15 +02:00
vit
[pytes collection] Fix flax test collection (#34004)
2024-10-07 18:11:13 +02:00
vit_mae
Fix ViT-MAE decoder interpolate (#33330)
2024-09-30 18:47:13 +02:00
vit_msn
Rename test_model_common_attributes -> test_model_get_set_embeddings (#31321)
2024-06-07 19:40:26 +01:00
vitdet
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
vitmatte
Update kwargs validation for preprocess with decorator (#32024)
2024-08-06 11:33:05 +01:00
vitpose
Add ViTPose (#30530)
2025-01-08 16:02:14 +00:00
vitpose_backbone
Add ViTPose (#30530)
2025-01-08 16:02:14 +00:00
vits
Generate tests: modality-agnostic input preparation (#33685)
2024-10-03 14:01:24 +01:00
vivit
Add sdpa for Vivit (#33757)
2024-10-15 11:27:54 +02:00
wav2vec2
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
wav2vec2_bert
Uniformize model processors (#31368)
2024-10-02 10:41:08 +02:00
wav2vec2_conformer
Remove trust_remote_code when loading Libri Dummy (#31748)
2024-07-23 14:54:38 +08:00
wav2vec2_phoneme
clean_up_tokenization_spaces=False if unset (#31938)
2024-09-26 19:38:20 +02:00
wav2vec2_with_lm
Pass datasets trust_remote_code (#31406)
2024-06-17 17:29:13 +01:00
wavlm
Remove trust_remote_code when loading Libri Dummy (#31748)
2024-07-23 14:54:38 +08:00
whisper
[Whisper] 🚨 Fix whisper decoding 🚨 (#34135)
2024-12-18 14:13:21 +01:00
x_clip
Load sub-configs from composite configs (#34410)
2024-11-05 11:34:01 +01:00
xglm
avoid calling gc.collect and cuda.empty_cache (#34514)
2024-10-31 16:36:13 +01:00
xlm
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
xlm_roberta
[RoBERTa-based] Add support for sdpa (#30510)
2024-08-28 10:26:00 +02:00
xlm_roberta_xl
Tests: upgrade test_eager_matches_sdpa_generate (#34386)
2024-10-25 11:55:07 +01:00
xlnet
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
xmod
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
yolos
[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)
2024-12-13 14:33:45 +01:00
yoso
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
zamba
Zamba new attention standard (#35375)
2025-01-07 10:08:45 +01:00
zoedepth
Add post_process_depth_estimation to image processors and support ZoeDepth's inference intricacies (#32550)
2024-10-22 15:50:54 +02:00
__init__.py
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00