riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 02:02:21 +06:00
Code Issues Actions 33 Packages Projects Releases Wiki Activity
50d3530aa0
transformers/tests/models
History
Ryan Mullins 50d3530aa0
Gemma3 (#36658) 
* Fix converter

* [Broken] Adds Gemma 3 to Hugging Face Transformers

* Consolidating Config and Processor params across impls

* Sorting out configuration parameters. Adds qk_norm before RoPE. Still not sure if RoPE is right.

* Additional plumbing for CausalLM and ConditionalGeneration variants

* incomplete draft of Orbax conversion script

* More complete checkpoint conversion

* Supporting Gemma 3 1B checkpoints

* Updating RoPE for multiple frequencies

* Adjustments to rotary embedder

* Proof of life for text-only operation

* Updating the conversion script to handle multimodal projection weights

* Fixing tet-only conversions

* Cleaner conversion script with multimodal support and a simpler processor

* Additional refatcors to the Gemma3Processor

* Simplified Processor to work over text representations

* Updated conversion script to join text and vision embeddings at converion time

* Logging for debugging

* Update src/transformers/models/gemma2/modeling_gemma2.py

Co-authored-by: Joshua Lochner <admin@xenova.com>

* Removed extraneous Config params

* Switching to fast tokenizer for checkpoint conversions

* isolating siglip for performance tetsing

* Minor changes for debugging tests against baselines

* Adding average pooling for soft tokens

* Updating processor code to enable simpler embedding interleaving for arbitrary number of images in prompts

* Updating conversion script for ShieldGemma 2 conversion compatibility

* Allow disable_compile to be provided as a kwarg

* Refresh from modular

* Updated conversion script and corrected sliding window

* Fix type mismatch in cache_position (#4)

* Fix dtype (#5)

* Fix type mismatch in cache_position

* Actually fix in the modular file

Co-authored-by: Aritra Roy Gosthipaty <aritra.born2fly@gmail.com>

---------

Co-authored-by: Aritra Roy Gosthipaty <aritra.born2fly@gmail.com>

* fixes for embedding table overflow and missing image_soft_token_mask from Gemma3Processor

* Adding 2D pooling for image embeddings

* Revert "Adding 2D pooling for image embeddings"

This reverts commit 65350cf531.

* Gemma3 average pooling changed from 1D to 2D

* Major refactor to Gemma3MultimodalInputProjection

* Updating Gemm 3 Auto* registrations

* Add option to save Gemma 3 chat template with tokenizer during weights conversion

* Removing unused imports

* Moving out-of-vocab handling from Gemma3Processor to Gemma3ForConditionalGeneration

* Removing duplicate config property

* Removing final logit softcapping and 1-indexing of position ids

* Fixing image processor config and none --> None typo

* Fixing sliding window size for 1B

* Updating image_mean and image_std in Image Processor

* Attention masking changed to lower triangular

* Moving image special tokens to conversion script

* Mirror image processor defaults from conversion script into Gemma3ProcessorKwargs

* Remove special token variables from symbol space

* Moving image soft token mask computation from Gemma3Processor to Gemma3ForConditionalGeneration

* tie lm_head and embedding weights

Co-authored-by: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com>

* Correct tied weights in Gemma3CausalLM

* iterative bidirectional attention

* resolving merge conflicts

* Reverting to Gemma 2 HybridCache with sldiing window support and a sliding_window_pattern of 6

* Correcting RoPE scaling

* clean up first pass, dummy model geenration works

* final clean up before fixing tests

* causal lm test works, so fine

* Fix conversion

* Update src/transformers/models/gemma3/processing_gemma3.py

* model tests are happy

* processor tests are happy

* image processing tests added

* fixup

* Fix pre-processing in conversion

* Inputs merging

* Do not normalize vision embeddings

* Apply Ryan's (and team) changes to attention

* token type ids + mask

* template

* move embed scale, add rope scale, fix tests

* Add chat template to tokenizer

* Use prefix for causal model loading

* use existing code for sliding mask from gemma2

* self.embed_tokens already normalizes

* Correcting Gemma3TextConfig parameters in conversion script

* typo, modular overwrites my fixes

* enable device map for text model

* Conversion updates

* ultra nit: no einsums

* update image token

* copy deepcopy config + some docs

* add some test, still WIP

* Refactoring --include_chat_tempalte logic in converter

* Update src/transformers/models/gemma3/modular_gemma3.py

Co-authored-by: Xuan-Son Nguyen <thichthat@gmail.com>

* Add eos tokens for instruct models

* dump so i can work on dgx

* Removing add_bos by default

* dump

* add fast im proc

* docs for PaS + fixup

* another fixup

* one more fixup

* fix tests

* Inverting prior BOS change

* ultra nit

* Reverting to Tokenizer saved with add_bos_token=True and chat template starting with BOS

* resize embeds, remove sqrt, add slow test outputs

* FA2 but quality is meh

* nit

* skip FA2, no idea what happened

* last bit for green CI

* please, green CI for docs

* T_T

* Fix for Gemma3 logits

* Support both options for system prompt

* Update src/transformers/models/gemma3/image_processing_gemma3_fast.py

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

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

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

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

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

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

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

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

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

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

Co-authored-by: Pedro Cuenca <pedro@huggingface.co>

* Docs updates now that assets are live

* Style fixes

---------

Co-authored-by: Joshua Lochner <admin@xenova.com>
Co-authored-by: Pedro Cuenca <pedro@huggingface.co>
Co-authored-by: Aritra Roy Gosthipaty <aritra.born2fly@gmail.com>
Co-authored-by: Mayank Chaturvedi <imayank@google.com>
Co-authored-by: Matthew Douglas <38992547+matthewdouglas@users.noreply.github.com>
Co-authored-by: raushan <raushan@huggingface.co>
Co-authored-by: Raushan Turganbay <raushan.turganbay@alumni.nu.edu.kz>
Co-authored-by: Xuan-Son Nguyen <thichthat@gmail.com>
Co-authored-by: Lysandre <hi@lysand.re>
2025-03-12 09:06:17 +01:00
..
albert [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
align Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
altclip Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
aria Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 [tests] remove pt_tf equivalence tests (#36253) 2025-02-19 11:55:11 +00:00
autoformer Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
aya_vision Add aya (#36521) 2025-03-04 12:24:33 +01:00
bamba Fix bamba tests amd (#36535) 2025-03-04 15:24:27 +01:00
bark CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
bart Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
bert CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
bert_generation CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
bigbird_pegasus Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
biogpt CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
bit Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +00:00
blenderbot [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +00:00
blenderbot_small [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +00:00
blip Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
blip_2 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
bloom [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +00:00
bridgetower [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
bros CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
chameleon CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
chinese_clip Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
clap Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
clip Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
clipseg Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
clvp [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
code_llama Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
codegen CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
cohere CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
cohere2 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
colpali Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
conditional_detr [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +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 [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
ctrl CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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 [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +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 [tests] remove flax-pt equivalence and cross tests (#36283) 2025-02-19 15:13:27 +00:00
dbrx CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
deberta [tests] remove pt_tf equivalence tests (#36253) 2025-02-19 11:55:11 +00:00
deberta_v2 [tests] remove pt_tf equivalence tests (#36253) 2025-02-19 11:55:11 +00:00
decision_transformer [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
deformable_detr [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
deit Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
detr [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
diffllama CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
dinat Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +00:00
dinov2 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
dinov2_with_registers Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
efficientnet Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +00:00
electra [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
emu3 🔴 VLM: compile compatibility (#35724) 2025-02-14 15:23:49 +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 [tests] remove flax-pt equivalence and cross tests (#36283) 2025-02-19 15:13:27 +00:00
ernie CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
esm [tests] fix EsmModelIntegrationTest::test_inference_bitsandbytes (#36225) 2025-02-17 11:10:33 +01:00
falcon [tests] enable bnb tests on xpu (#36233) 2025-02-24 11:30:15 +01:00
falcon_mamba [docs] Redesign (#31757) 2025-03-03 10:33:46 -08: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 [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
flava Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
fnet Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
focalnet Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +00:00
fsmt [tests] remove test_export_to_onnx (#36241) 2025-02-17 16:52:44 +00: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 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
gemma Gemma3 (#36658) 2025-03-12 09:06:17 +01:00
gemma2 Gemma3 (#36658) 2025-03-12 09:06:17 +01:00
gemma3 Gemma3 (#36658) 2025-03-12 09:06:17 +01:00
git Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
glm CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01: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 Fast image processor and refactor slow one (#36185) 2025-03-01 00:56:00 -05:00
gpt_bigcode CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
gpt_neo [tests] remove flax-pt equivalence and cross tests (#36283) 2025-02-19 15:13:27 +00:00
gpt_neox CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
gpt_neox_japanese CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
gpt_sw3 🚨 No more default chat templates (#31733) 2024-07-24 17:36:32 +01:00
gpt2 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
gptj [tests] remove flax-pt equivalence and cross tests (#36283) 2025-02-19 15:13:27 +00:00
granite CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
granitemoe CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
granitemoeshared add shared experts for upcoming Granite 4.0 language models (#35894) 2025-02-14 16:55:28 +01:00
grounding_dino [GroundingDino] Fix grounding dino loss 🚨 (#31828) 2025-02-27 19:15:58 +00:00
groupvit Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
helium CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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 [tests] remove pt_tf equivalence tests (#36253) 2025-02-19 11:55:11 +00:00
ibert [warnings] fix E721 warnings (#32223) 2024-07-25 15:12:23 +02:00
idefics Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
idefics2 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
idefics3 SmolVLM2 (#36126) 2025-02-20 15:00:26 +01:00
ijepa Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +00:00
imagegpt Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
informer Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
instructblip Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
instructblipvideo Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
jamba CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
jetmoe CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
kosmos2 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
layoutlm Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
layoutlmv2 [tests] remove pt_tf equivalence tests (#36253) 2025-02-19 11:55:11 +00:00
layoutlmv3 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
layoutxlm [tests] remove pt_tf equivalence tests (#36253) 2025-02-19 11:55:11 +00:00
led CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
levit [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
lilt Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
llama [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +00:00
llava Refactor some core stuff (#36539) 2025-03-11 09:26:28 +01:00
llava_next Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
llava_next_video Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
llava_onevision Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
luke Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
lxmert Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
m2m_100 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
mamba Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
mamba2 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
marian Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
markuplm [tests] remove pt_tf equivalence tests (#36253) 2025-02-19 11:55:11 +00: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 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
mbart50 Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
megatron_bert [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00: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 [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +00:00
mixtral CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
mllama Fix bugs in mllama image processing (#36156) 2025-03-11 10:22:48 +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 [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
moonshine [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
moshi CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
mra Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
mt5 [tests] remove test_export_to_onnx (#36241) 2025-02-17 16:52:44 +00:00
musicgen [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
musicgen_melody [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
mvp CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
myt5 [Red CIs] Fix hub failures (#34001) 2024-10-07 10:56:24 +02:00
nemotron CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
nllb Skip tests properly (#31308) 2024-06-26 21:59:08 +01:00
nllb_moe Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
olmo2 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
olmoe CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
opt [tests] remove tf/flax tests in /generation (#36235) 2025-02-17 14:59:22 +00:00
owlv2 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
owlvit Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
paligemma Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
paligemma2 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
patchtsmixer CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
patchtst Add missing atol to torch.testing.assert_close where rtol is specified (#36234) 2025-02-17 14:57:50 +01:00
pegasus Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
pegasus_x [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +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 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
phi CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
phi3 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
phimoe [generate] remove cache v4.47 deprecations (#36212) 2025-02-17 13:55:03 +00:00
phobert Adds pretrained IDs directly in the tests (#29534) 2024-03-13 14:53:27 +01:00
pix2struct Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
pixtral Refactoring of ImageProcessorFast (#35069) 2025-02-04 17:52:31 -05:00
plbart Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
poolformer Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +00:00
pop2piano [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
prophetnet CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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 Revert qwen2 breaking changes related to attention refactor (#36162) 2025-02-14 13:44:14 +01:00
qwen2_5_vl [generate] remove cache v4.47 deprecations (#36212) 2025-02-17 13:55:03 +00:00
qwen2_audio [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
qwen2_moe CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
qwen2_vl CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
rag [tests] make cuda-only tests device-agnostic (#35222) 2024-12-18 10:14:22 +01:00
recurrent_gemma [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
reformer Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
regnet Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +00:00
rembert [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
resnet Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +00:00
roberta CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
roberta_prelayernorm CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
roc_bert [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
roformer Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
rt_detr Au revoir flaky test_fast_is_faster_than_slow (#36240) 2025-02-17 18:30:07 +01: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 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
sam Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
seamless_m4t Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
seamless_m4t_v2 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
segformer [tests] remove pt_tf equivalence tests (#36253) 2025-02-19 11:55:11 +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
siglip2 Add SigLIP 2 (#36323) 2025-02-21 09:04:19 +00:00
smolvlm Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
speech_encoder_decoder [tests] remove flax-pt equivalence and cross tests (#36283) 2025-02-19 15:13:27 +00:00
speech_to_text Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
speecht5 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
starcoder2 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
superglue CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
superpoint CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +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 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 typos in tests (#36547) 2025-03-05 15:04:06 -08:00
t5 [tests] remove flax-pt equivalence and cross tests (#36283) 2025-02-19 15:13:27 +00:00
table_transformer [CI] Check test if the GenerationTesterMixin inheritance is correct 🐛 🔫 (#36180) 2025-02-21 10:18:20 +00:00
tapas [tests] remove pt_tf equivalence tests (#36253) 2025-02-19 11:55:11 +00:00
textnet Fix tests for vision models (#35654) 2025-02-13 10:28:37 +00:00
time_series_transformer Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
tvp [tests] further fix Tester object has no attribute '_testMethodName' (#35781) 2025-01-29 16:05:33 +01:00
udop Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
umt5 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
videomae Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +00:00
vilt Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
vipllava Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
vision_encoder_decoder [tests] remove flax-pt equivalence and cross tests (#36283) 2025-02-19 15:13:27 +00:00
vision_text_dual_encoder [tests] remove flax-pt equivalence and cross tests (#36283) 2025-02-19 15:13:27 +00:00
visual_bert Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
vit Fix tests for vision models (#35654) 2025-02-13 10:28:37 +00:00
vit_mae [tests] remove pt_tf equivalence tests (#36253) 2025-02-19 11:55:11 +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 Fixed VitDet for non-squre Images (#35969) 2025-02-25 19:31:24 +00:00
vitmatte Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
wav2vec2_bert Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
wav2vec2_conformer Fix typos in tests (#36547) 2025-03-05 15:04:06 -08: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 Add missing atol to torch.testing.assert_close where rtol is specified (#36234) 2025-02-17 14:57:50 +01:00
whisper Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
x_clip Fix typos in tests (#36547) 2025-03-05 15:04:06 -08:00
xglm [XGLM] tag tests as slow (#36592) 2025-03-06 17:54:41 +00:00
xlm CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01: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 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
xlnet CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
xmod CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
yolos Add common test for torch.export and fix some vision models (#35124) 2025-02-11 11:37:31 +00:00
yoso CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
zamba CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
zamba2 CI: avoid human error, automatically infer generative models (#33212) 2025-02-13 16:27:11 +01:00
zoedepth Fix tests for vision models (#35654) 2025-02-13 10:28:37 +00:00
__init__.py Move test model folders (#17034) 2022-05-03 14:42:02 +02:00