Yoni Gozlan 2b46943195 Add GOT-OCR 2.0 to Transformers (#34721) ...

* init modular got_ocr2

* Get correct got_ocr architecture

* add processing

* run modular with processing

* add working inference

* apply modular

* Refactor and fix style

* Refactor, cleanup, fix style

* fix init order

* Fix docs

* add base modeling tests

* fix style and consistency

* rename doc file

* fix repo consistency

* fix inference with box

* add image processing and support for crop_to_multi_page

* Fix batch inference

* add tests

* fixup

* fix slow test

* fix docstrings

* Add model doc

* update to new init

* fix input autocast pixel_values dtype

* update doc

* move doc to multimodal

* Reformat crop_image_to_patches and add docstrings

* Fix example in forward docstring

* Address Pablo review

* [run slow] got_ocr2

* remove defaults defined twice

* apply modular

* add torch_device to integration tests

* update modular

* follow-up Pavel review

* add device variable in doc

* fix doc multi-page

* Force eager attention for vision encoder to avoid attn implementation conflict

* revert qwen2vl doc changes

* use Qwen2ForCausalLM instead of Qwen2Model

* make fixup

* refactor gotocr2 to llava style

* uniformize function names and reduce checks

* final nits

* fix pixel_values dtype error

* change checkpoint names

* fix modular

2025-01-31 11:28:13 -05: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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

bamba

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

blenderbot

Forbid PretrainedConfig from saving generate parameters; Update deprecations in generate-related code 🧹 (#32659)

2024-08-23 11:12:53 +01:00

blenderbot_small

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

blip

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

blip_2

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +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

Fix mask slicing for models with HybridCache (#35681)

2025-01-28 14:35:00 +01:00

colpali

Add support for nested images to LLava and VipLLava (#35558)

2025-01-30 16:49:20 -05:00

conditional_detr

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

convnextv2

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

deit

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

depth_anything

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

detr

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

dinov2

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

dinov2_with_registers

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

efficientnet

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

electra

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

emu3

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

[tests] make cuda-only tests device-agnostic (#35607)

2025-01-13 14:48:39 +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

Fix mask slicing for models with HybridCache (#35681)

2025-01-28 14:35:00 +01:00

git

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

glm

Fix some tests (#35682)

2025-01-17 12:10:43 +00:00

glpn

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

gpt_neo

Generation tests: update imagegpt input name, remove unused functions (#33663)

2024-09-24 16:40:48 +01:00

gpt_neox

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

Test: generate with torch.compile(model.forward) as a fast test (#34544)

2025-01-28 14:10:38 +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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

imagegpt

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

instructblipvideo

Add support for nested images to LLava and VipLLava (#35558)

2025-01-30 16:49:20 -05: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

Pixtral: vectorize patch embeddings and enable tests (#35122)

2025-01-30 12:40:18 +01:00

llava_next

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

llava_next_video

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

llava_onevision

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

maskformer

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

mobilenet_v2

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

mobilevit

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

mobilevitv2

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

pixtral

Add support for nested images to LLava and VipLLava (#35558)

2025-01-30 16:49:20 -05:00

plbart

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01:00

poolformer

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01: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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

pvt_v2

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

add qwen2.5vl (#35569)

2025-01-23 11:23:00 +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

Add GOT-OCR 2.0 to Transformers (#34721)

2025-01-31 11:28:13 -05:00

rag

[tests] make cuda-only tests device-agnostic (#35222)

2024-12-18 10:14:22 +01:00

recurrent_gemma

Tests: move generate tests to the right mixin and delete redundant tests (#34464)

2024-10-30 10:59:08 +00:00

reformer

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

regnet

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01: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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01: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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

seggpt

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

speech_encoder_decoder

🚨All attention refactor🚨 (#35235)

2024-12-18 16:53:39 +01:00

speech_to_text

[tests] fix "Tester object has no attribute '_testMethodName'" (#34910)

2024-12-13 14:33:45 +01:00

speecht5

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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

swin

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

swin2sr

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

swinv2

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

video_llava

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

videomae

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01: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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

vit_mae

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

vit_msn

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

vitdet

Skip tests properly (#31308)

2024-06-26 21:59:08 +01:00

vitmatte

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

vitpose

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

vitpose_backbone

Add ViTPose (#30530)

2025-01-08 16:02:14 +00: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

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01: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

Whisper: fix static cache CI (#35852)

2025-01-30 12:43:00 +01: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

Make pipeline able to load processor (#32514)

2024-10-09 16:46: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

xlnet

Make pipeline able to load processor (#32514)

2024-10-09 16:46:11 +01: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

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01: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

Add Zamba2 (#34517)

2025-01-27 10:51:23 +01:00

zoedepth

[tests] further fix Tester object has no attribute '_testMethodName' (#35781)

2025-01-29 16:05:33 +01:00

__init__.py

Move test model folders (#17034)

2022-05-03 14:42:02 +02:00