Skip to content

Commit 237140d

Browse files
authored
Shorter names for cerebras and OVH
1 parent cb8aa46 commit 237140d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

llm_benchmark_suite.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -243,7 +243,7 @@ class _OvhLlm(_Llm):
243243
def __init__(self, model: str, display_model: Optional[str] = None):
244244
super().__init__(
245245
"",
246-
"endpoints.ai.cloud.ovh.net/" + display_model,
246+
"cloud.ovh.net/" + display_model,
247247
base_url=f"https://{model}.endpoints.kepler.ai.cloud.ovh.net/api/openai_compat/v1",
248248
)
249249

@@ -254,7 +254,7 @@ class _CerebrasLlm(_Llm):
254254
def __init__(self, model: str, display_model: Optional[str] = None):
255255
super().__init__(
256256
model,
257-
"api.cerebras.ai/" + (display_model or model),
257+
"cerebras.ai/" + (display_model or model),
258258
api_key=os.getenv("CEREBRAS_API_KEY"),
259259
base_url="https://api.cerebras.ai/v1",
260260
)

0 commit comments

Comments
 (0)