Skip to content

Commit

Permalink
update new provider doc
Browse files Browse the repository at this point in the history
  • Loading branch information
hanouticelina committed Feb 14, 2025
1 parent e679891 commit bc3f360
Showing 1 changed file with 2 additions and 0 deletions.
2 changes: 2 additions & 0 deletions src/huggingface_hub/inference/_providers/new_provider.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,8 @@ 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`.

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

Expand Down

0 comments on commit bc3f360

Please sign in to comment.