SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity
Files
71de20b818c3aa9715fb3d0e26f448ec534b03d2
HuggingFace_transformer/tests/models
History
Crystalcareai 71de20b818 Add Arcee model support (#38621) 
* Add Arcee model support to transformers

- Add ArceeConfig and model mappings for all task types (CausalLM, SequenceClassification, QuestionAnswering, TokenClassification)
- Add auto-loading support through AutoModel, AutoConfig, and AutoTokenizer
- Use LlamaTokenizer for tokenization
- Add FX graph support for Arcee models
- Create lazy loading module structure for Arcee

* feat: update YARN scaling and RoPE validation for Arcee model

* feat: add auto_docstring checkpoint config to Arcee model classes

* docs: add pre-trained model weights reference to Arcee configuration files

* refactor: move RoPE utilities to dedicated modeling_rope_utils module

* Add comprehensive test suite for Arcee model

- Add test_modeling_arcee.py following standard transformers test patterns
- Include tests for all model variants (CausalLM, SequenceClassification, QuestionAnswering, TokenClassification)
- Add specific test for ReLU² activation in ArceeMLP
- Add RoPE scaling tests including YARN support
- Follow CausalLMModelTest pattern used by similar models

* Add documentation for Arcee model

- Add comprehensive model documentation with usage examples
- Include all model variants in autodoc
- Add to table of contents in proper alphabetical order
- Fixes documentation coverage for Arcee model classes

* Make style/fixup

* fix copyright year

* Sync modular conversion

* revert in legacy supported models in src/transformers/utils/fx

* cleaned redundant code in modular_arcee.py

* cleaned testing

* removed pretraining tp

* fix styles

* integration testing

---------

Co-authored-by: Pranav <veldurthipranav@gmail.com>
Co-authored-by: Pranav <56645758+pranav4501@users.noreply.github.com>
2025-06-24 15:05:29 +02:00
..
albert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
align
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
altclip
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
arcee
Add Arcee model support (#38621)
2025-06-24 15:05:29 +02:00
aria
Don't run AriaForConditionalGenerationModelTest on CircleCI (#38615)
2025-06-06 11:30:31 +02:00
audio_spectrogram_transformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
auto
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
autoformer
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
aya_vision
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
bamba
enable misc test cases on XPU (#38852)
2025-06-18 09:20:49 +02:00
bark
enable more test cases on xpu (#38572)
2025-06-06 09:29:51 +02:00
bart
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
barthez
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
bartpho
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
beit
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
bert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
bert_generation
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
bert_japanese
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
bertweet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
big_bird
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
bigbird_pegasus
[generation] bring back tests on vision models (#38603)
2025-06-06 08:23:15 +00:00
biogpt
Bart: new cache format (#35314)
2025-05-16 13:26:54 +02:00
bit
Add ImageProcessorFast to BiT processor (#37180)
2025-04-14 17:07:48 +02:00
bitnet
Add Bitnet model (#37742)
2025-04-28 15:08:46 +02:00
blenderbot
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
blenderbot_small
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
blip
Fix more flaky test_initialization (#38932)
2025-06-20 17:28:32 +02:00
blip_2
[generation] bring back tests on vision models (#38603)
2025-06-06 08:23:15 +00:00
bloom
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
bridgetower
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
bros
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
byt5
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
camembert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
canine
Skip torchscript tests for 2 models (#38643)
2025-06-06 20:17:37 +02:00
chameleon
Update some tests for torch 2.7.1 (#38701)
2025-06-10 11:46:52 +02:00
chinese_clip
Add Fast Chinese-CLIP Processor (#37012)
2025-04-15 18:31:20 +02:00
clap
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
clip
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
clipseg
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
clvp
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
code_llama
remove unhandled parameter (#38145)
2025-06-02 15:57:32 +02:00
codegen
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
cohere
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
cohere2
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
colpali
Add ColQwen2 to 🤗 transformers (#35778)
2025-06-02 12:58:01 +00:00
colqwen2
Update some tests for torch 2.7.1 (#38701)
2025-06-10 11:46:52 +02:00
conditional_detr
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
convbert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
convnext
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
convnextv2
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
cpm
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
cpmant
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
csm
Update CsmForConditionalGenerationIntegrationTest (#38424)
2025-05-28 10:20:43 +02:00
ctrl
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
cvt
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
d_fine
enable misc test cases on XPU (#38852)
2025-06-18 09:20:49 +02:00
dab_detr
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
dac
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
data2vec
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
dbrx
Refactor DBRX tests to use CausalLMModelTest base classes (#38475)
2025-06-13 16:22:12 +01:00
deberta
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
deberta_v2
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
decision_transformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
deepseek_v3
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
deformable_detr
🚨🚨 Fix initialization of Mask2Former (#38864)
2025-06-18 09:46:22 +02:00
deit
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
depth_anything
Skip some export tests on torch 2.7 (#38677)
2025-06-12 12:47:15 +02:00
depth_pro
Fix more flaky test_initialization (#38932)
2025-06-20 17:28:32 +02:00
detr
enable more test cases on xpu (#38572)
2025-06-06 09:29:51 +02:00
diffllama
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
dinat
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
dinov2
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
dinov2_with_registers
Fix more flaky test_initialization (#38932)
2025-06-20 17:28:32 +02:00
distilbert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
dit
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
donut
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
dpr
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
dpt
Skip some tests for now (#38931)
2025-06-20 11:05:49 +02:00
efficientnet
Add EfficientNet Image PreProcessor (#37055)
2025-04-16 21:59:24 +02:00
electra
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
emu3
update emu3 test (#38543)
2025-06-03 11:02:01 +02:00
encodec
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
encoder_decoder
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
ernie
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
esm
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
falcon
🚨 🚨 Inherited CausalLM Tests (#37590)
2025-05-23 18:29:31 +01:00
falcon_h1
[Falcon H1] Fix slow path forward pass (#38320)
2025-05-26 15:30:35 +02:00
falcon_mamba
Fix FalconMambaIntegrationTests (#38566)
2025-06-19 13:50:33 +02:00
fastspeech2_conformer
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
flaubert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
flava
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
fnet
🚨 rm already deprecated pad_to_max_length arg (#37617)
2025-05-01 15:21:55 +02:00
focalnet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
fsmt
Fix fsmt tests (#38904)
2025-06-19 10:56:34 +02:00
funnel
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
fuyu
🔴 [VLM] Add base model without head (#37033)
2025-05-07 17:47:51 +02:00
gemma
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
gemma2
Unbreak optimum-executorch (#38646)
2025-06-13 11:13:32 +02:00
gemma3
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
git
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
glm
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
glm4
enable misc test cases on XPU (#38852)
2025-06-18 09:20:49 +02:00
glpn
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
got_ocr2
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
gpt2
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
gpt_bigcode
Remove head mask in generative models (#35786)
2025-05-15 10:44:19 +02:00
gpt_neo
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
gpt_neox
🚨 🚨 Inherited CausalLM Tests (#37590)
2025-05-23 18:29:31 +01:00
gpt_neox_japanese
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
gpt_sw3
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
gptj
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
granite
switch to device agnostic device calling for test cases (#38247)
2025-05-26 10:18:53 +02:00
granite_speech
enable finegrained_fp8 and granite_speech cases on XPU (#38036)
2025-05-14 08:58:40 +00:00
granitemoe
switch to device agnostic device calling for test cases (#38247)
2025-05-26 10:18:53 +02:00
granitemoehybrid
switch to device agnostic device calling for test cases (#38247)
2025-05-26 10:18:53 +02:00
granitemoeshared
switch to device agnostic device calling for test cases (#38247)
2025-05-26 10:18:53 +02:00
grounding_dino
enable more test cases on xpu (#38572)
2025-06-06 09:29:51 +02:00
groupvit
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
helium
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
herbert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
hgnet_v2
Add D-FINE Model into Transformers (#36261)
2025-04-29 12:17:55 +01:00
hiera
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
hubert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
ibert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
idefics
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
idefics2
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
idefics3
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
ijepa
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
imagegpt
[test] update test_past_key_values_format (#37614)
2025-04-22 11:07:34 +01:00
informer
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
instructblip
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
instructblipvideo
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
internvl
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
jamba
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
janus
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
jetmoe
🚨 🚨 Inherited CausalLM Tests (#37590)
2025-05-23 18:29:31 +01:00
kosmos2
[VLMs] support attention backends (#37576)
2025-05-08 18:18:54 +02:00
layoutlm
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
layoutlmv2
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
layoutlmv3
Skip some tests for now (#38931)
2025-06-20 11:05:49 +02:00
layoutxlm
🚨 rm already deprecated pad_to_max_length arg (#37617)
2025-05-01 15:21:55 +02:00
led
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
levit
Add Fast LeViT Processor (#37154)
2025-04-14 17:07:36 +02:00
lightglue
Add LightGlue model (#31718)
2025-06-17 18:10:23 +02:00
lilt
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
llama
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
llama4
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
llava
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
llava_next
Fix llava_next tests (#38813)
2025-06-13 15:19:41 +02:00
llava_next_video
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
llava_onevision
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
longformer
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
longt5
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
luke
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
lxmert
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
m2m_100
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
mamba
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mamba2
align xpu's autocast behavior w/ cuda by using device agnostic torch APIs (#38284)
2025-06-19 11:48:23 +00:00
marian
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
markuplm
No more Tuple, List, Dict (#38797)
2025-06-17 19:37:18 +01:00
mask2former
🚨🚨 Fix initialization of Mask2Former (#38864)
2025-06-18 09:46:22 +02:00
maskformer
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
mbart
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
mbart50
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
megatron_bert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
megatron_gpt2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mgp_str
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mimi
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
minimax
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
mistral
fix mistral and mistral3 tests (#38978)
2025-06-23 17:07:18 +02:00
mistral3
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
mixtral
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
mlcd
Add MLCD model (#36182)
2025-04-15 11:33:09 +01:00
mllama
Fix mllama (#38704)
2025-06-12 16:15:35 +02:00
mluke
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
mobilebert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
mobilenet_v1
Add Fast Image Processor for MobileNetV1 (#37111)
2025-04-23 15:55:41 -04:00
mobilenet_v2
Add Fast Mobilenet-V2 Processor (#37113)
2025-04-14 17:08:47 +02:00
mobilevit
Skip some tests for now (#38931)
2025-06-20 11:05:49 +02:00
mobilevitv2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
modernbert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
moonshine
Skip torchscript tests for 2 models (#38643)
2025-06-06 20:17:37 +02:00
moshi
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
mpnet
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
mpt
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
mra
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
mt5
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
musicgen
enable misc test cases on XPU (#38852)
2025-06-18 09:20:49 +02:00
musicgen_melody
enable misc test cases on XPU (#38852)
2025-06-18 09:20:49 +02:00
mvp
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
myt5
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
nemotron
switch to device agnostic device calling for test cases (#38247)
2025-05-26 10:18:53 +02:00
nllb
Use lru_cache for tokenization tests (#36818)
2025-03-28 15:09:35 +01:00
nllb_moe
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
nougat
Skip some tests for now (#38931)
2025-06-20 11:05:49 +02:00
nystromformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
olmo
Unbreak optimum-executorch (#38646)
2025-06-13 11:13:32 +02:00
olmo2
Make HF implementation match original OLMo 2 models for lower precisions (#38131)
2025-05-19 15:35:23 +02:00
olmoe
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
omdet_turbo
enable more test cases on xpu (#38572)
2025-06-06 09:29:51 +02:00
oneformer
Fix OneFormer integration test (#38016)
2025-05-12 16:02:41 +02:00
openai
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
opt
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
owlv2
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
owlvit
Add Fast owlvit Processor (#37164)
2025-04-14 17:58:09 +02:00
paligemma
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
paligemma2
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
patchtsmixer
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
patchtst
Force torch>=2.6 with torch.load to avoid vulnerability issue (#37785)
2025-04-25 16:57:09 +02:00
pegasus
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
pegasus_x
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
perceiver
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
persimmon
🚨 🚨 Inherited CausalLM Tests (#37590)
2025-05-23 18:29:31 +01:00
phi
🚨 🚨 Inherited CausalLM Tests (#37590)
2025-05-23 18:29:31 +01:00
phi3
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
phi4_multimodal
Fix phi4_multimodal tests (#38816)
2025-06-18 09:39:17 +02:00
phimoe
Fix MoE gradient test (#38438)
2025-05-28 16:44:20 +01:00
phobert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
pix2struct
Fix more flaky test_initialization (#38932)
2025-06-20 17:28:32 +02:00
pixtral
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
plbart
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
poolformer
Add Fast Image Processor for PoolFormer (#37182)
2025-04-23 15:55:33 -04:00
pop2piano
Fix typos in strings and comments (#37799)
2025-04-28 11:39:11 +01:00
prompt_depth_anything
Skip some export tests on torch 2.7 (#38677)
2025-06-12 12:47:15 +02:00
prophetnet
[generation] bring back tests on vision models (#38603)
2025-06-06 08:23:15 +00:00
pvt
Add Fast PVT Processor (#37204)
2025-04-23 15:55:20 -04:00
pvt_v2
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
qwen2
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
qwen2_5_omni
Fix qwen_2_5 omni (#38658)
2025-06-12 14:43:54 +02:00
qwen2_5_vl
Fix qwen2_5_vl tests (#38845)
2025-06-17 10:55:24 +02:00
qwen2_audio
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
qwen2_moe
Fix MoE gradient test (#38438)
2025-05-28 16:44:20 +01:00
qwen2_vl
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
qwen3
Fix qwen3 tests (#38862)
2025-06-17 15:21:36 +02:00
qwen3_moe
Fix qwen3_moe tests (#38865)
2025-06-18 14:36:03 +02:00
rag
Fix rag (#38585)
2025-06-23 17:42:46 +02:00
recurrent_gemma
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
reformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
regnet
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
rembert
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
resnet
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
roberta
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
roberta_prelayernorm
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
roc_bert
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
roformer
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
rt_detr
enable more test cases on xpu (#38572)
2025-06-06 09:29:51 +02:00
rt_detr_v2
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
rwkv
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
sam
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
sam_hq
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
seamless_m4t
[seamless_m4t] Skip some tests when speech is not available (#38430)
2025-06-02 09:17:28 +00:00
seamless_m4t_v2
[seamless_m4t] Skip some tests when speech is not available (#38430)
2025-06-02 09:17:28 +00:00
segformer
Skip some tests for now (#38931)
2025-06-20 11:05:49 +02:00
seggpt
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
sew
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
sew_d
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
shieldgemma2
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
siglip
[tests] expand flex-attn test for vision models (#38434)
2025-06-03 07:40:44 +00:00
siglip2
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
smolvlm
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
speech_encoder_decoder
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
speech_to_text
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
speecht5
[generation] bring back tests on vision models (#38603)
2025-06-06 08:23:15 +00:00
splinter
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
squeezebert
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
stablelm
🚨 🚨 Inherited CausalLM Tests (#37590)
2025-05-23 18:29:31 +01:00
starcoder2
🚨 🚨 Inherited CausalLM Tests (#37590)
2025-05-23 18:29:31 +01:00
superglue
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
superpoint
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
swiftformer
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
swin
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
swin2sr
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
swinv2
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
switch_transformers
[generation] bring back tests on vision models (#38603)
2025-06-06 08:23:15 +00:00
t5
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
table_transformer
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
tapas
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
textnet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
time_series_transformer
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
timesfm
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
timesformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
timm_backbone
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
timm_wrapper
Add kwargs for timm.create_model in TimmWrapper (#38860)
2025-06-20 12:00:09 +00:00
trocr
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
tvp
Add Optional to remaining types (#37808)
2025-04-28 14:20:45 +01:00
udop
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
umt5
[generation] bring back tests on vision models (#38603)
2025-06-06 08:23:15 +00:00
unispeech
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
unispeech_sat
🔴🔴🔴 [Attention] Refactor Attention Interface for Bart-based Models (#38108)
2025-05-22 17:12:58 +02:00
univnet
No more Tuple, List, Dict (#38797)
2025-06-17 19:37:18 +01:00
upernet
Skip some export tests on torch 2.7 (#38677)
2025-06-12 12:47:15 +02:00
video_llava
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
videomae
[tests] expand flex-attn test for vision models (#38434)
2025-06-03 07:40:44 +00:00
vilt
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
vipllava
[tests] expand flex-attn test for vision models (#38434)
2025-06-03 07:40:44 +00:00
vision_encoder_decoder
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
vision_text_dual_encoder
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
visual_bert
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
vit
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
vit_mae
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
vit_msn
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
vitdet
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
vitmatte
Add Idefics2/3 and SmolVLM Fast image processors + improvements for fast image processors (#38157)
2025-06-23 14:17:25 +00:00
vitpose
Skip some export tests on torch 2.7 (#38677)
2025-06-12 12:47:15 +02:00
vitpose_backbone
Remove dead protected imports (#38980)
2025-06-23 13:44:50 +02:00
vits
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
vivit
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
vjepa2
Add V-JEPA for video classification model (#38788)
2025-06-13 17:56:15 +01:00
wav2vec2
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
wav2vec2_bert
🚨 🚨 Setup -> setupclass conversion (#37282)
2025-04-08 17:15:37 +01:00
wav2vec2_conformer
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
wav2vec2_phoneme
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +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 Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
whisper
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
x_clip
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
xglm
Expectation fixes and added AMD expectations (#38729)
2025-06-13 16:14:58 +02:00
xlm
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
xlm_roberta
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
xlm_roberta_xl
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
xlnet
Deprecate TF + JAX (#38758)
2025-06-11 17:28:06 +01:00
xmod
Remove old code for PyTorch, Accelerator and tokenizers (#37234)
2025-04-10 20:54:21 +02:00
yolos
🚨Early-error🚨 config will error out if output_attentions=True and the attn implementation is wrong (#38288)
2025-05-23 17:17:38 +02:00
yoso
Use Python 3.9 syntax in tests (#37343)
2025-04-08 14:12:08 +02:00
zamba
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
zamba2
Remove all traces of low_cpu_mem_usage (#38792)
2025-06-12 16:39:33 +02:00
zoedepth
Skip some export tests on torch 2.7 (#38677)
2025-06-12 12:47:15 +02:00
__init__.py
…