Cerebras Integration

This commit is contained in:
Henry Tu 2024-11-20 10:20:28 -08:00
parent 34be07e0df
commit 3838bd1704
16 changed files with 515 additions and 65 deletions

View file

@ -94,6 +94,7 @@ class TestInference:
"remote::tgi",
"remote::together",
"remote::fireworks",
"remote::cerebras",
):
pytest.skip("Other inference providers don't support completion() yet")
@ -139,6 +140,7 @@ class TestInference:
"remote::tgi",
"remote::together",
"remote::fireworks",
"remote::cerebras",
):
pytest.skip(
"Other inference providers don't support structured output in completions yet"