Sarah Ciston commited on
Commit
cbe9258
·
1 Parent(s): 3b8ec11

change hyperparam

Browse files
Files changed (1) hide show
  1. sketch.js +3 -3
sketch.js CHANGED
@@ -30,12 +30,12 @@ var PROMPT = `The [BLANK] works as a [FILL] but wishes for [FILL].`
30
  // let out = await pipe(PREPROMPT + PROMPT)
31
  let out = await pipe(PREPROMPT + PROMPT, {
32
  max_new_tokens: 250,
33
- temperature: 2,
34
  // return_full_text: False,
35
- repetition_penalty: 1.5,
36
  // no_repeat_ngram_size: 2,
37
  // num_beams: 2,
38
- num_return_sequences: 2
39
  });
40
  console.log(out)
41
 
 
30
  // let out = await pipe(PREPROMPT + PROMPT)
31
  let out = await pipe(PREPROMPT + PROMPT, {
32
  max_new_tokens: 250,
33
+ temperature: 0.9,
34
  // return_full_text: False,
35
+ repetition_penalty: 1.5
36
  // no_repeat_ngram_size: 2,
37
  // num_beams: 2,
38
+ // num_return_sequences: 2
39
  });
40
  console.log(out)
41