diff --git a/src/huggingface_hub/inference/_providers/__init__.py b/src/huggingface_hub/inference/_providers/__init__.py index cbac62a786..bfaaaa9113 100644 --- a/src/huggingface_hub/inference/_providers/__init__.py +++ b/src/huggingface_hub/inference/_providers/__init__.py @@ -29,6 +29,7 @@ HFInferenceTask, ) from .hyperbolic import HyperbolicTextGenerationTask, HyperbolicTextToImageTask +from .latitude import LatitudeConversationalTask, LatitudeTextGenerationTask from .nebius import ( NebiusConversationalTask, NebiusFeatureExtractionTask, @@ -73,6 +74,7 @@ "groq", "hf-inference", "hyperbolic", + "latitude-sh", "nebius", "novita", "nscale", @@ -156,6 +158,10 @@ "conversational": HyperbolicTextGenerationTask("conversational"), "text-generation": HyperbolicTextGenerationTask("text-generation"), }, + "latitude-sh": { + "conversational": LatitudeConversationalTask(), + "text-generation": LatitudeTextGenerationTask(), + }, "nebius": { "text-to-image": NebiusTextToImageTask(), "conversational": NebiusConversationalTask(), diff --git a/src/huggingface_hub/inference/_providers/latitude.py b/src/huggingface_hub/inference/_providers/latitude.py new file mode 100644 index 0000000000..b6cb88aede --- /dev/null +++ b/src/huggingface_hub/inference/_providers/latitude.py @@ -0,0 +1,11 @@ +from ._common import BaseConversationalTask, BaseTextGenerationTask + + +class LatitudeConversationalTask(BaseConversationalTask): + def __init__(self): + super().__init__(provider="latitude-sh", base_url="https://api.lsh.ai") + + +class LatitudeTextGenerationTask(BaseTextGenerationTask): + def __init__(self): + super().__init__(provider="latitude-sh", base_url="https://api.lsh.ai")