Skip to content

Commit

Permalink
move to root level
Browse files Browse the repository at this point in the history
  • Loading branch information
radames committed Mar 30, 2024
1 parent 65a3f84 commit b755faf
Showing 1 changed file with 4 additions and 6 deletions.
10 changes: 4 additions & 6 deletions packages/inference/test/HfInference.spec.ts
Original file line number Diff line number Diff line change
Expand Up @@ -655,12 +655,10 @@ describe.concurrent(
const res = await ep.textGeneration({
model: "tgi",
messages: [{ role: "user", content: "Complete the this sentence with words one plus one is equal " }],
parameters: {
max_tokens: 500,
return_full_text: false,
temperature: 0.0,
seed: 0,
},
max_tokens: 500,
return_full_text: false,
temperature: 0.1,
seed: 0,
});
if (res.choices && res.choices.length > 0) {
const completion = res.choices[0].message?.content;
Expand Down

0 comments on commit b755faf

Please sign in to comment.