Skip to content

Commit

Permalink
[docs] Improve register_pipeline (#35300)
Browse files Browse the repository at this point in the history
register_pipeline
  • Loading branch information
stevhliu authored Dec 17, 2024
1 parent 77080f0 commit 0531d75
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion docs/source/en/add_new_pipeline.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
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
from pair_classification import PairClassificationPipeline
Expand Down

0 comments on commit 0531d75

Please sign in to comment.