From 8467da136ecb9d6965c89ad006cefbea8efad32e Mon Sep 17 00:00:00 2001 From: Celina Hanouti Date: Fri, 14 Feb 2025 15:29:43 +0100 Subject: [PATCH] update provider doc --- src/huggingface_hub/inference/_providers/new_provider.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/huggingface_hub/inference/_providers/new_provider.md b/src/huggingface_hub/inference/_providers/new_provider.md index 3e6bf421fe..48f2988863 100644 --- a/src/huggingface_hub/inference/_providers/new_provider.md +++ b/src/huggingface_hub/inference/_providers/new_provider.md @@ -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