Skip to content

Commit

Permalink
update provider doc
Browse files Browse the repository at this point in the history
  • Loading branch information
hanouticelina committed Feb 14, 2025
1 parent 3381ced commit 8467da1
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/huggingface_hub/inference/_providers/new_provider.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@ Implement the methods that require custom handling. Check out the base implement

If the provider supports multiple tasks that require different implementations, create dedicated subclasses for each task, following the pattern shown in `fal_ai.py`.

For `text-generation` and `conversational` tasks, one can just inherit from `BaseTextGenerationTask` (defined in `_common.py`) and override the methods if needed. Examples can be found in `hyperbolic.py` and `sambanova.py`.
For `text-generation` and `conversational` tasks, one can just inherit from `BaseTextGenerationTask` and `BaseConversationalTask` respectively (defined in `_common.py`) and override the methods if needed. Examples can be found in `fireworks_ai.py` and `together.py`.

```py
from typing import Any, Dict, Optional, Union
Expand Down

0 comments on commit 8467da1

Please sign in to comment.