transformers/utils/test_module
Sylvain Gugger dc9147ff36
Custom pipeline (#18079)
* Initial work

* More work

* Add tests for custom pipelines on the Hub

* Protect import

* Make the test work for TF as well

* Last PyTorch specific bit

* Add documentation

* Style

* Title in toc

* Bad names!

* Update docs/source/en/add_new_pipeline.mdx

Co-authored-by: Lysandre Debut <lysandre.debut@reseau.eseo.fr>

* Auto stash before merge of "custom_pipeline" and "origin/custom_pipeline"

* Address review comments

* Address more review comments

* Update src/transformers/pipelines/__init__.py

Co-authored-by: Lysandre Debut <lysandre.debut@reseau.eseo.fr>

Co-authored-by: Lysandre Debut <lysandre.debut@reseau.eseo.fr>
2022-07-19 12:02:35 +02:00
..
__init__.py Save code of registered custom models (#15379) 2022-02-02 10:44:37 -05:00
custom_configuration.py Make sure custom configs work with Transformers (#15569) 2022-02-09 10:04:44 -05:00
custom_feature_extraction.py Custom feature extractor (#15630) 2022-02-11 16:43:54 -05:00
custom_modeling.py Fix from_pretrained with default base_model_prefix (#15814) 2022-02-24 11:43:51 +01:00
custom_pipeline.py Custom pipeline (#18079) 2022-07-19 12:02:35 +02:00
custom_processing.py Allow custom code for Processors (#15649) 2022-02-15 09:44:35 -05:00
custom_tokenization_fast.py Save code of registered custom models (#15379) 2022-02-02 10:44:37 -05:00
custom_tokenization.py Save code of registered custom models (#15379) 2022-02-02 10:44:37 -05:00