From c2a9c79b2c0bc98cb50fa145a188fe7854d48ffe Mon Sep 17 00:00:00 2001
From: Kai Huang <kaihuang724@gmail.com>
Date: Mon, 10 Feb 2025 20:02:18 -0800
Subject: [PATCH] draft: edited hyperbolic model names

---
 packages/inference/test/HfInference.spec.ts | 14 ++++++--------
 1 file changed, 6 insertions(+), 8 deletions(-)

diff --git a/packages/inference/test/HfInference.spec.ts b/packages/inference/test/HfInference.spec.ts
index 0d9bf3d33..fda580a4d 100644
--- a/packages/inference/test/HfInference.spec.ts
+++ b/packages/inference/test/HfInference.spec.ts
@@ -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 () => {
@@ -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" });
 			});