Skip to content

Commit

Permalink
draft: edited hyperbolic model names
Browse files Browse the repository at this point in the history
  • Loading branch information
Kaihuang724 committed Feb 11, 2025
1 parent 179c84d commit c2a9c79
Showing 1 changed file with 6 additions and 8 deletions.
14 changes: 6 additions & 8 deletions packages/inference/test/HfInference.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -1185,7 +1185,7 @@ describe.concurrent("HfInference", () => {
"meta-llama/Llama-3.2-3B-Instruct": "meta-llama/Llama-3.2-3B-Instruct",
"meta-llama/Llama-3.3-70B-Instruct": "meta-llama/Llama-3.3-70B-Instruct",
"stabilityai/stable-diffusion-2": "stabilityai/stable-diffusion-2",
"meta-llama/Llama-3.1-405B-BASE-FP8": "meta-llama/Llama-3.1-405B-BASE-FP8",
"meta-llama/Llama-3.1-405B": "meta-llama/Llama-3.1-405B",
};

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

it("textGeneration", async () => {
const res = await client.textGeneration({
model: "meta-llama/Llama-3.1-405B-BASE-FP8",
model: "meta-llama/Llama-3.1-405B",
provider: "hyperbolic",
inputs: "Paris is",
parameters: {
temperature: 0,
max_tokens: 10,
},
messages: [{ role: "user", content: "Paris is" }],
temperature: 0,
max_tokens: 10,
});
expect(res).toMatchObject({ generated_text: " city of love" });
});
Expand Down

0 comments on commit c2a9c79

Please sign in to comment.