transformers/tests/models/clip
Pedro Cuenca cb8e3ee25f
Add FlaxCLIPTextModelWithProjection (#25254)
* Add FlaxClipTextModelWithProjection

This is necessary to support the Flax port of Stable Diffusion XL: fb6d705fb5/text_encoder_2/config.json (L3)

Co-authored-by: Martin Müller <martin.muller.me@gmail.com>
Co-authored-by: Juan Acevedo <juancevedo@gmail.com>

* Use FlaxCLIPTextModelOutput

* make fix-copies again

* Apply suggestions from code review

Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

* Use `return_dict` for consistency with other uses.

Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>

* Fix docstring example.

* Add new model to FlaxCLIPTextModelTest

* Add to IGNORE_NON_AUTO_CONFIGURED list

* Fix naming convention.

---------

Co-authored-by: Martin Müller <martin.muller.me@gmail.com>
Co-authored-by: Juan Acevedo <juancevedo@gmail.com>
Co-authored-by: Sanchit Gandhi <93869735+sanchit-gandhi@users.noreply.github.com>
2023-08-25 10:58:14 +02:00
..
__init__.py
test_image_processing_clip.py Refactor image processor testers (#25450) 2023-08-11 11:30:18 +01:00
test_modeling_clip.py Fix `test_model_parallelism` (#25359) 2023-08-08 10:48:45 +02:00
test_modeling_flax_clip.py Add FlaxCLIPTextModelWithProjection (#25254) 2023-08-25 10:58:14 +02:00
test_modeling_tf_clip.py Speed up TF tests by reducing hidden layer counts (#24595) 2023-06-30 16:30:33 +01:00
test_processor_clip.py
test_tokenization_clip.py fix: Text splitting in the BasicTokenizer (#22280) 2023-07-11 11:07:58 -04:00