Skip to content

Commit c2a9c79

Browse files
committed
draft: edited hyperbolic model names
1 parent 179c84d commit c2a9c79

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

packages/inference/test/HfInference.spec.ts

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1185,7 +1185,7 @@ describe.concurrent("HfInference", () => {
11851185
"meta-llama/Llama-3.2-3B-Instruct": "meta-llama/Llama-3.2-3B-Instruct",
11861186
"meta-llama/Llama-3.3-70B-Instruct": "meta-llama/Llama-3.3-70B-Instruct",
11871187
"stabilityai/stable-diffusion-2": "stabilityai/stable-diffusion-2",
1188-
"meta-llama/Llama-3.1-405B-BASE-FP8": "meta-llama/Llama-3.1-405B-BASE-FP8",
1188+
"meta-llama/Llama-3.1-405B": "meta-llama/Llama-3.1-405B",
11891189
};
11901190

11911191
it("chatCompletion - hyperbolic", async () => {
@@ -1227,20 +1227,18 @@ describe.concurrent("HfInference", () => {
12271227
const res = await client.textToImage({
12281228
model: "stabilityai/stable-diffusion-2",
12291229
provider: "hyperbolic",
1230-
inputs: "award winning high resolution photo of a giant tortoise",
1230+
messages: [{ role: "user", content: "award winning high resolution photo of a giant tortoise" }],
12311231
});
12321232
expect(res).toBeInstanceOf(Blob);
12331233
});
12341234

12351235
it("textGeneration", async () => {
12361236
const res = await client.textGeneration({
1237-
model: "meta-llama/Llama-3.1-405B-BASE-FP8",
1237+
model: "meta-llama/Llama-3.1-405B",
12381238
provider: "hyperbolic",
1239-
inputs: "Paris is",
1240-
parameters: {
1241-
temperature: 0,
1242-
max_tokens: 10,
1243-
},
1239+
messages: [{ role: "user", content: "Paris is" }],
1240+
temperature: 0,
1241+
max_tokens: 10,
12441242
});
12451243
expect(res).toMatchObject({ generated_text: " city of love" });
12461244
});

0 commit comments

Comments
 (0)