Skip to content

Commit 59cf309

Browse files
authored
test: fix a test (#473)
1 parent 9cdbce9 commit 59cf309

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test/modelDependent/llama3.2/promptCompletion.test.ts

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,9 +41,9 @@ describe("llama 3.2", () => {
4141
});
4242

4343
const promptCompletion = await chatSession.completePrompt("Hi there!", {
44-
maxTokens: 50
44+
maxTokens: 11
4545
});
46-
expect(promptCompletion).toMatchInlineSnapshot("\" I'm looking for a new phone case. I need a case that can protect your phone from scratches and drops.\"");
46+
expect(promptCompletion).toMatchInlineSnapshot(`" I'm looking for a new phone case. I need"`);
4747
expect(LlamaText.fromTokens(model.tokenizer, chatSession.sequence.contextTokens)).toMatchInlineSnapshot(`
4848
LlamaText([
4949
new SpecialToken("BOS"),
@@ -64,7 +64,7 @@ describe("llama 3.2", () => {
6464
new SpecialTokensText("<|end_header_id|>"),
6565
"
6666
67-
Hi there! I'm looking for a new phone case. I need a case that can protect your phone from scratches and drops.",
67+
Hi there! I'm looking for a new phone case. I",
6868
])
6969
`);
7070

0 commit comments

Comments
 (0)