Skip to content

Commit

Permalink
increase timeout
Browse files Browse the repository at this point in the history
  • Loading branch information
ksyeo1010 committed Sep 24, 2024
1 parent 1924e53 commit addfe69
Showing 1 changed file with 5 additions and 5 deletions.
10 changes: 5 additions & 5 deletions binding/nodejs/test/index.test.ts
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@ import {

import * as testData from '../../../resources/.test/test_data.json';

jest.setTimeout(30000);
jest.setTimeout(60000);

const ACCESS_KEY = process.argv
.filter(x => x.startsWith('--access_key='))[0]
Expand Down Expand Up @@ -340,9 +340,9 @@ describe('PicoLLM generate tests', () => {
device: DEVICE
});

const numPromptTokens = (picoLLM.tokenize(prompt, true, false)).length;

try {
const numPromptTokens = (picoLLM.tokenize(prompt, true, false)).length;

const res = await picoLLM.generate(prompt, {
completionTokenLimit: completionTokenLimit,
seed: seeds[0],
Expand Down Expand Up @@ -394,9 +394,9 @@ describe('PicoLLM generate tests', () => {
device: DEVICE
});

const numPromptTokens = (picoLLM.tokenize(prompt, true, false)).length;

try {
const numPromptTokens = (picoLLM.tokenize(prompt, true, false)).length;

const res = await picoLLM.generate(prompt, {
completionTokenLimit: completionTokenLimit,
seed: seed,
Expand Down

0 comments on commit addfe69

Please sign in to comment.