Skip to content

Commit ee91c9f

Browse files
committed
test: reduce flakiness
1 parent 6cf55bf commit ee91c9f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

test/modelDependent/llama3.1/tokenPredictor.test.ts

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ import {compareTokens} from "../../../src/utils/compareTokens.js";
66

77
describe("llama 3.1", () => {
88
describe("token predictor", () => {
9-
test("DraftModelTokenPredictor", {timeout: 1000 * 60 * 60 * 2}, async () => {
9+
test("DraftModelTokenPredictor", {retry: 4, timeout: 1000 * 60 * 60 * 2}, async () => {
1010
const modelPath = await getModelFile("Meta-Llama-3.1-8B-Instruct.Q4_K_M.gguf");
1111
const llama = await getTestLlama();
1212

0 commit comments

Comments
 (0)