Skip to content

Commit ecb5988

Browse files
committed
fix: fix Novita test case
1 parent 819a358 commit ecb5988

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

packages/inference/test/HfInference.spec.ts

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1131,6 +1131,10 @@ describe.concurrent("HfInference", () => {
11311131
() => {
11321132
const client = new HfInference(env.HF_NOVITA_KEY);
11331133

1134+
HARDCODED_MODEL_ID_MAPPING["novita"] = {
1135+
"meta-llama/llama-3.1-8b-instruct": "meta-llama/llama-3.1-8b-instruct",
1136+
};
1137+
11341138
it("chatCompletion", async () => {
11351139
const res = await client.chatCompletion({
11361140
model: "meta-llama/llama-3.1-8b-instruct",

0 commit comments

Comments
 (0)