transformers/tests/models/flava
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 [feat] Add FLAVA model (#16654) 2022-05-11 14:56:48 -07:00
test_image_processing_flava.py Input data format (#25464) 2023-08-16 17:45:02 +01:00
test_modeling_flava.py CI with num_hidden_layers=2 🚀🚀🚀 (#25266) 2023-08-02 20:22:36 +02:00
test_processor_flava.py Vision processors - replace FE with IPs (#20590) 2022-12-09 10:48:34 +00:00