sarahciston commited on
Commit
a244abf
1 Parent(s): 468bd4a

remove some hyperparams

Browse files
Files changed (1) hide show
  1. sketch.js +6 -6
sketch.js CHANGED
@@ -17,13 +17,13 @@ async function textGenTask(input){
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
- min_new_tokens: 100,
24
- no_repeat_ngram_size: 2,
25
- num_beams: 2,
26
- num_return_sequences: 2
27
  })
28
 
29
  console.log(await out)
 
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
+ min_new_tokens: 100
24
+ // no_repeat_ngram_size: 2,
25
+ // num_beams: 2,
26
+ // num_return_sequences: 2
27
  })
28
 
29
  console.log(await out)