saymrwulf/transformers
1
0
Fork 0
mirror of https://github.com/saymrwulf/transformers.git synced 2026-05-14 20:58:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[docs] Improve register_pipeline (#35300)

Browse source 
register_pipeline
This commit is contained in:
Steven Liu 2024-12-17 10:27:23 -08:00 committed by GitHub
parent 77080f023f
commit 0531d7513b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
docs/source/en/add_new_pipeline.md
View file

@ -184,7 +184,7 @@ class PairClassificationPipeline(Pipeline):
``` ```
The implementation is framework agnostic, and will work for PyTorch and TensorFlow models. If we have saved this in The implementation is framework agnostic, and will work for PyTorch and TensorFlow models. If we have saved this in
a file named `pair_classification.py`, we can then import it and register it like this: a file named `pair_classification.py`, we can then import it and register it like this. The [register_pipeline](https://github.com/huggingface/transformers/blob/9feae5fb0164e89d4998e5776897c16f7330d3df/src/transformers/pipelines/base.py#L1387) function registers the pipeline details (task type, pipeline class, supported backends) to a models `config.json` file.
```py ```py
from pair_classification import PairClassificationPipeline from pair_classification import PairClassificationPipeline