Skip to content

Commit 4c06878

Browse files
committed
test: fix tests
1 parent db4a243 commit 4c06878

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/modelDependent/llama3.1/tokenBias.test.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,18 +24,18 @@ describe("llama 3.1", () => {
2424
for (const token of model.iterateAllTokens()) {
2525
const text = model.detokenize([token]);
2626

27-
if (text.toLowerCase().includes("hello"))
27+
if (text.toLowerCase().includes("well"))
2828
customBias.set(token, -0.99);
2929
else if (text.toLowerCase().includes("hi"))
3030
customBias.set(token, "never");
3131
}
3232

33-
const res = await chatSession.prompt('Greet me by saying "hello" to me', {
33+
const res = await chatSession.prompt('Greet me by saying "hi" to me', {
3434
tokenBias: customBias,
3535
maxTokens: 100
3636
});
3737

38-
expect(res.toLowerCase()).to.not.include("hello");
38+
expect(res.toLowerCase()).to.not.include("well");
3939
expect(res.toLowerCase()).to.not.include("hi ");
4040
expect(res.toLowerCase()).to.not.include("hi.");
4141
});

0 commit comments

Comments
 (0)