transformers/tests/models/pix2struct
amyeroberts 6bca43bb90
Input data format (#25464)
* Add copied from statements for image processors

* Move out rescale and normalize to base image processor

* Remove rescale and normalize from vit (post rebase)

* Update docstrings and tidy up

* PR comments

* Add input_data_format as preprocess argument

* Resolve tests and tidy up

* Remove num_channels argument

* Update doc strings -> default ints not in code formatting
2023-08-16 17:45:02 +01:00
..
__init__.py Add Pix2Struct (#21400) 2023-03-22 16:53:52 +01:00
test_image_processing_pix2struct.py Input data format (#25464) 2023-08-16 17:45:02 +01:00
test_modeling_pix2struct.py CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
test_processor_pix2struct.py Pix2StructImageProcessor requires torch>=1.11.0 (#24270) 2023-06-14 17:05:40 +02:00