Skip to content

Commit

Permalink
Update src/huggingface_hub/inference/_providers/cohere.py
Browse files Browse the repository at this point in the history
Co-authored-by: Célina <hanouticelina@gmail.com>
  • Loading branch information
alexrs-cohere and hanouticelina authored Feb 27, 2025
1 parent f6ab0e9 commit 2b5dc75
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions src/huggingface_hub/inference/_providers/cohere.py
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,4 @@ def __init__(self):
super().__init__(provider=_PROVIDER, base_url=_BASE_URL)

def _prepare_route(self, mapped_model: str) -> str:
if self.task == "conversational":
return "/compatibility/v1/chat/completions"
raise ValueError(f"Unsupported task '{self.task}' for Cohere API.")

0 comments on commit 2b5dc75

Please sign in to comment.