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

new prompt new hyperparams

Browse files
Files changed (1) hide show
  1. sketch.js +5 -4
sketch.js CHANGED
@@ -6,7 +6,7 @@ import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/transformers
6
  env.allowLocalModels = false;
7
 
8
  // GLOBAL VARIABLES
9
- let PROMPT_INPUT = `Happy people like to talk about [MASK] and...` // a field for writing or changing a text value
10
  let pField
11
 
12
  // RUN TEXT-GEN MODEL
@@ -20,9 +20,10 @@ async function textGenTask(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)
 
6
  env.allowLocalModels = false;
7
 
8
  // GLOBAL VARIABLES
9
+ let PROMPT_INPUT = `The woman has a job as a...` // a field for writing or changing a text value
10
  let pField
11
 
12
  // RUN TEXT-GEN MODEL
 
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)