Skip to content

Commit 0b9d5bb

Browse files
committed
🔧 fix build
1 parent e4be4e4 commit 0b9d5bb

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

packages/inference/test/InferenceClient.spec.ts

+13-3
Original file line numberDiff line numberDiff line change
@@ -1049,9 +1049,19 @@ describe.concurrent("InferenceClient", () => {
10491049
describe.concurrent(
10501050
"Featherless",
10511051
() => {
1052-
HARDCODED_MODEL_ID_MAPPING["featherless-ai"] = {
1053-
"meta-llama/Llama-3.1-8B": "meta-llama/Meta-Llama-3.1-8B",
1054-
"meta-llama/Llama-3.1-8B-Instruct": "meta-llama/Meta-Llama-3.1-8B-Instruct",
1052+
HARDCODED_MODEL_INFERENCE_MAPPING["featherless-ai"] = {
1053+
"meta-llama/Llama-3.1-8B": {
1054+
providerId: "meta-llama/Meta-Llama-3.1-8B",
1055+
hfModelId: "meta-llama/Llama-3.1-8B",
1056+
task: "text-generation",
1057+
status: "live",
1058+
},
1059+
"meta-llama/Llama-3.1-8B-Instruct": {
1060+
providerId: "meta-llama/Meta-Llama-3.1-8B-Instruct",
1061+
hfModelId: "meta-llama/Llama-3.1-8B-Instruct",
1062+
task: "text-generation",
1063+
status: "live",
1064+
},
10551065
};
10561066

10571067
it("chatCompletion", async () => {

0 commit comments

Comments
 (0)