transformers/tests/models/vivit
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 ViViT (#22518) 2023-07-11 14:04:04 +01:00
test_image_processing_vivit.py Input data format (#25464) 2023-08-16 17:45:02 +01:00
test_modeling_vivit.py Update tiny model info. and pipeline testing (#25213) 2023-07-31 19:35:33 +02:00