sarahciston commited on
Commit
a1e1d61
1 Parent(s): 8df278e

only params to control length

Browse files
Files changed (1) hide show
  1. sketch.js +5 -4
sketch.js CHANGED
@@ -20,10 +20,11 @@ async function textGenTask(input){
20
  // temperature: 2,
21
  // no_repeat_ngram_size: 2,
22
  // num_beams: 2,
23
- // num_return_sequences: 2
24
- max_new_tokens: 200,
25
- min_new_tokens: 100,
26
- repetition_penalty: 2
 
27
  })
28
 
29
  console.log(await out)
 
20
  // temperature: 2,
21
  // no_repeat_ngram_size: 2,
22
  // num_beams: 2,
23
+ // num_return_sequences: 2,
24
+ // repetition_penalty: 2,
25
+ // min_new_tokens: 100,
26
+ min_new_tokens: 50,
27
+ max_new_tokens: 200
28
  })
29
 
30
  console.log(await out)