SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
e60ae0d0784d7fc5b35eb69b0e5e5e35aab5e70e
HuggingFace_transformer/tests/models
History
Sambhav Dixit 950cfb0b4f Fix PaliGemma Pad Token Masking During Training #35855 (#35859) 
* change order of unmasking of tokens

* library import

* class setup

* test function

* refactor

* add commit message

* test modified

* explict initiliasation of weights + made model smaller

* removed sepete testing file

* fixup

* fixup core

* test attention mask with token types

* tests fixup

* removed PaliGemmaAttentionMaskTest class

---------

Co-authored-by: sambhavnoobcoder <indosambahv@gmail.com>
2025-02-13 10:11:44 +01:00
..
albert
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
align
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
altclip
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
aria
VLM: enable skipped tests (#35746)
2025-02-12 12:55:46 +01:00
audio_spectrogram_transformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +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
AutoformerForPrediction test add atol (#36017)
2025-02-10 19:22:24 +01:00
bamba
Fix CI issues (#35662)
2025-02-11 18:17:01 +01:00
bark
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
bart
[bart] minor test fixes (#35965)
2025-01-30 10:00:11 +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 common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
bert
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
bert_generation
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +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
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
bigbird_pegasus
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
biogpt
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
bit
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00: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
Refactoring of ImageProcessorFast (#35069)
2025-02-04 17:52:31 -05:00
blip_2
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
bloom
VLMs: enable generation tests (#33533)
2024-09-19 12:04:24 +02:00
bridgetower
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
bros
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
byt5
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
camembert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
canine
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
chameleon
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
chinese_clip
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
clap
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
clip
Refactoring of ImageProcessorFast (#35069)
2025-02-04 17:52:31 -05:00
clipseg
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
clvp
Iterative generation using Input embeds and past_key_values (#35890)
2025-02-06 11:06:05 +01:00
code_llama
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
codegen
[generate] move max time tests (#35962)
2025-01-29 17:56:46 +00:00
cohere
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
cohere2
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
colpali
Add support for nested images to LLava and VipLLava (#35558)
2025-01-30 16:49:20 -05:00
conditional_detr
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
convbert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
convnext
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
convnextv2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
cpm
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
cpmant
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
ctrl
avoid calling gc.collect and cuda.empty_cache (#34514)
2024-10-31 16:36:13 +01:00
cvt
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
dab_detr
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
dac
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
data2vec
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
dbrx
Test: generate with torch.compile(model.forward) as a fast test (#34544)
2025-01-28 14:10:38 +00:00
deberta
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
deberta_v2
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
decision_transformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
deformable_detr
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
deit
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
depth_anything
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
depth_pro
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
detr
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
diffllama
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
dinat
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
dinov2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
dinov2_with_registers
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
distilbert
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
dit
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
donut
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
dpr
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
dpt
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
efficientnet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
electra
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
emu3
Chat template: update for processor (#35953)
2025-02-10 09:52:19 +01:00
encodec
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +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
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
falcon
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
falcon_mamba
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
fastspeech2_conformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
flaubert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
flava
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
fnet
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
focalnet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
fsmt
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
funnel
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
fuyu
Iterative generation using Input embeds and past_key_values (#35890)
2025-02-06 11:06:05 +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
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
git
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
glm
Fix some tests (#35682)
2025-01-17 12:10:43 +00:00
glpn
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
got_ocr2
Add GOT-OCR 2.0 to Transformers (#34721)
2025-01-31 11:28:13 -05:00
gpt2
[generate] move max time tests (#35962)
2025-01-29 17:56:46 +00:00
gpt_bigcode
Iterative generation using Input embeds and past_key_values (#35890)
2025-02-06 11:06:05 +01:00
gpt_neo
Generation tests: update imagegpt input name, remove unused functions (#33663)
2024-09-24 16:40:48 +01:00
gpt_neox
Refactor (and fix) gpt_neox (#35610)
2025-02-04 11:18:43 +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
[generate] move max time tests (#35962)
2025-01-29 17:56:46 +00:00
granite
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
granitemoe
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
grounding_dino
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
groupvit
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
helium
Add-helium (#35669)
2025-01-13 18:41:15 +01:00
herbert
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
hiera
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
hubert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
ibert
[warnings] fix E721 warnings (#32223)
2024-07-25 15:12:23 +02:00
idefics
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
idefics2
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
idefics3
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
ijepa
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
imagegpt
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
informer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
instructblip
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
instructblipvideo
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
jamba
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
jetmoe
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
kosmos2
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
layoutlm
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
layoutlmv2
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
layoutlmv3
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
layoutxlm
Fix #34494 assistant tokens when truncated (#34531)
2024-11-05 15:10:15 +00:00
led
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
levit
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
lilt
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
llama
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
llava
Chat template: update for processor (#35953)
2025-02-10 09:52:19 +01:00
llava_next
Chat template: update for processor (#35953)
2025-02-10 09:52:19 +01:00
llava_next_video
Refactoring of ImageProcessorFast (#35069)
2025-02-04 17:52:31 -05:00
llava_onevision
Chat template: update for processor (#35953)
2025-02-10 09:52:19 +01:00
longformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
longt5
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
luke
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
lxmert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
m2m_100
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
mamba
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
mamba2
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +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
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
mask2former
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
maskformer
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
mbart
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +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
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
mimi
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
mistral
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
mixtral
[tests] make cuda-only tests device-agnostic (#35607)
2025-01-13 14:48:39 +01:00
mllama
VLM: enable skipped tests (#35746)
2025-02-12 12:55:46 +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
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
mobilenet_v2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
mobilevit
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
mobilevitv2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
modernbert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
moonshine
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
moshi
VLM: enable skipped tests (#35746)
2025-02-12 12:55:46 +01:00
mpnet
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
mpt
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
mra
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +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
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
musicgen_melody
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
mvp
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
myt5
[Red CIs] Fix hub failures (#34001)
2024-10-07 10:56:24 +02:00
nemotron
[tests] make cuda-only tests device-agnostic (#35607)
2025-01-13 14:48:39 +01:00
nllb
Skip tests properly (#31308)
2024-06-26 21:59:08 +01:00
nllb_moe
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
nougat
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
nystromformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
olmo
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
olmo2
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
olmoe
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
omdet_turbo
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
oneformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +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
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
owlvit
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
paligemma
Fix PaliGemma Pad Token Masking During Training #35855 (#35859)
2025-02-13 10:11:44 +01:00
paligemma2
VLM: enable skipped tests (#35746)
2025-02-12 12:55:46 +01:00
patchtsmixer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
patchtst
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +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
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
perceiver
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
persimmon
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
phi
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
phi3
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
phimoe
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
phobert
Adds pretrained IDs directly in the tests (#29534)
2024-03-13 14:53:27 +01:00
pix2struct
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
pixtral
Refactoring of ImageProcessorFast (#35069)
2025-02-04 17:52:31 -05:00
plbart
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
poolformer
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
pop2piano
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
prophetnet
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
pvt
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
pvt_v2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
qwen2
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
qwen2_5_vl
Chat template: update for processor (#35953)
2025-02-10 09:52:19 +01:00
qwen2_audio
[Qwen2Audio] handle input ids expansion during processing (#35534)
2025-01-07 16:47:27 +01:00
qwen2_moe
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
qwen2_vl
Chat template: update for processor (#35953)
2025-02-10 09:52:19 +01:00
rag
[tests] make cuda-only tests device-agnostic (#35222)
2024-12-18 10:14:22 +01:00
recurrent_gemma
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
reformer
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
regnet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
rembert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
resnet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
roberta
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
roberta_prelayernorm
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
roc_bert
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
roformer
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
rt_detr
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
rt_detr_v2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
rwkv
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
sam
Handle empty change indices in SAM's mask to rle conversion (#35665)
2025-01-30 19:08:38 +00:00
seamless_m4t
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
seamless_m4t_v2
Make test_generate_with_static_cache even less flaky (#34995)
2024-12-20 16:03:26 +01:00
segformer
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
seggpt
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
sew
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
sew_d
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
siglip
Refactoring of ImageProcessorFast (#35069)
2025-02-04 17:52:31 -05:00
speech_encoder_decoder
🚨All attention refactor🚨 (#35235)
2024-12-18 16:53:39 +01:00
speech_to_text
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
speecht5
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
splinter
Make pipeline able to load processor (#32514)
2024-10-09 16:46:11 +01:00
squeezebert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
stablelm
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
starcoder2
[tests] make cuda-only tests device-agnostic (#35607)
2025-01-13 14:48:39 +01:00
superglue
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
superpoint
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
swiftformer
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
swin
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
swin2sr
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
swinv2
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
switch_transformers
Fix flaky SwitchTransformersModelTest::test_training_gradient (#35587)
2025-01-09 15:36:22 +01:00
t5
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
table_transformer
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
tapas
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
textnet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
time_series_transformer
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
timesformer
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
timm_backbone
Less flaky for TimmBackboneModelTest::test_batching_equivalence (#35971)
2025-01-30 16:56:26 +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
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +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
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
unispeech_sat
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
univnet
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
upernet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
video_llava
Refactoring of ImageProcessorFast (#35069)
2025-02-04 17:52:31 -05:00
videomae
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
vilt
[tests] further fix Tester object has no attribute '_testMethodName' (#35781)
2025-01-29 16:05:33 +01:00
vipllava
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
vision_encoder_decoder
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
vision_text_dual_encoder
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
visual_bert
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
vit
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
vit_mae
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
vit_msn
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
vitdet
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
vitmatte
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
vitpose
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
vitpose_backbone
skip test_initialization for VitPoseBackboneModelTest for now (#36154)
2025-02-12 18:24:24 +01:00
vits
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
vivit
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
wav2vec2
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
wav2vec2_bert
Byebye test_batching_equivalence's flakiness (#35729)
2025-01-21 13:11:33 +01:00
wav2vec2_conformer
Byebye test_batching_equivalence's flakiness (#35729)
2025-01-21 13:11:33 +01:00
wav2vec2_phoneme
clean_up_tokenization_spaces=False if unset (#31938)
2024-09-26 19:38:20 +02:00
wav2vec2_with_lm
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
wavlm
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
whisper
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
x_clip
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
xglm
[generate] move max time tests (#35962)
2025-01-29 17:56:46 +00:00
xlm
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
xlm_roberta
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
xlm_roberta_xl
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
xlnet
[generate] shape checks in tests compatible with fixed-length caches (+ some minor fixes) (#35993)
2025-02-10 17:50:54 +00:00
xmod
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
yolos
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
yoso
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
zamba
use torch.testing.assertclose instead to get more details about error in cis (#35659)
2025-01-24 16:55:28 +01:00
zamba2
Iterative generation using Input embeds and past_key_values (#35890)
2025-02-06 11:06:05 +01:00
zoedepth
Add common test for torch.export and fix some vision models (#35124)
2025-02-11 11:37:31 +00:00
__init__.py
Move test model folders (#17034)
2022-05-03 14:42:02 +02:00