sarahciston commited on
Commit
0dc072e
·
verified ·
1 Parent(s): f73cfe9

add hyperparams

Browse files
Files changed (1) hide show
  1. sketch.js +8 -1
sketch.js CHANGED
@@ -16,7 +16,14 @@ async function textGenTask(input){
16
 
17
  const pipe = await pipeline('text-generation', 'Xenova/LaMini-Cerebras-256M')
18
 
19
- var out = await pipe(input)
 
 
 
 
 
 
 
20
 
21
  console.log(await out)
22
  console.log('text-gen task completed')
 
16
 
17
  const pipe = await pipeline('text-generation', 'Xenova/LaMini-Cerebras-256M')
18
 
19
+ var out = await pipe(input, {
20
+ temperature: 2,
21
+ max_new_tokens: 200,
22
+ repetition_penalty: 2,
23
+ // no_repeat_ngram_size: 2,
24
+ // num_beams: 2,
25
+ // num_return_sequences: 2,
26
+ })
27
 
28
  console.log(await out)
29
  console.log('text-gen task completed')