transformers/tests/models/donut
Vinh H. Pham 7cc9e61a3a
Add Fast Image Processor for Donut (#37081)
* add donut fast image processor support

* run make style

* Update src/transformers/models/donut/image_processing_donut_fast.py

Co-authored-by: Parteek <parteekkamboj112@gmail.com>

* update test, remove none default values

* add do_align_axis = True test, fix bug in slow image processor

* run make style

* remove np usage

* make style

* Apply suggestions from code review

* Update src/transformers/models/donut/image_processing_donut_fast.py

Co-authored-by: Yoni Gozlan <74535834+yonigozlan@users.noreply.github.com>

* add size revert in preprocess

* make style

* fix copies

* add test for preprocess with kwargs

* make style

* handle None input_data_format in align_long_axis

---------

Co-authored-by: Parteek <parteekkamboj112@gmail.com>
Co-authored-by: Yoni Gozlan <74535834+yonigozlan@users.noreply.github.com>
2025-04-14 16:24:01 +02:00
..
__init__.py Add Donut (#18488) 2022-08-12 16:40:58 +02:00
test_image_processing_donut.py Add Fast Image Processor for Donut (#37081) 2025-04-14 16:24:01 +02:00
test_modeling_donut_swin.py Add image classifier donut & update loss calculation for all swins (#37224) 2025-04-10 15:00:42 +02:00
test_processor_donut.py 🚨 🚨 Setup -> setupclass conversion (#37282) 2025-04-08 17:15:37 +01:00