Sarah Ciston commited on
Commit
50b0541
·
1 Parent(s): d59354e

update preprompt nix hyperparams

Browse files
Files changed (1) hide show
  1. sketch.js +5 -5
sketch.js CHANGED
@@ -135,7 +135,7 @@ new p5(function (p5) {
135
 
136
  // BLANKS = inputValues // get ready to feed array list into model
137
 
138
- let PREPROMPT = `In the sample sentence I provide, please fill in the [BLANK] with each word I in the array ${BLANKSVALUES}, replace any [FILL] with an appropriate word of your choice, and respond with an array of sentences. Here is the SAMPLE SENTENCE: `
139
 
140
  // we pass PROMPT and PREPROMPT to the model function, don't need to pass BLANKSVALUES bc it's passed into the PREPROMPT already here
141
 
@@ -170,10 +170,10 @@ async function runModel(PREPROMPT, PROMPT){
170
  // , 'Xenova/distilgpt2'
171
 
172
  let out = await pipe(inputText, {
173
- max_tokens: 250,
174
- return_full_text: false,
175
- repetition_penalty: 1.5,
176
- num_return_sequences: 1 //must be 1 for greedy search
177
  })
178
 
179
  // let out = await pipe(inputText)
 
135
 
136
  // BLANKS = inputValues // get ready to feed array list into model
137
 
138
+ let PREPROMPT = `In the sentence I provide, please fill in the [BLANK] with each word in the array ${BLANKSVALUES}, replace any [FILL] with a word of your choice. Here is the SAMPLE SENTENCE: `
139
 
140
  // we pass PROMPT and PREPROMPT to the model function, don't need to pass BLANKSVALUES bc it's passed into the PREPROMPT already here
141
 
 
170
  // , 'Xenova/distilgpt2'
171
 
172
  let out = await pipe(inputText, {
173
+ max_tokens: 250
174
+ // return_full_text: false
175
+ // repetition_penalty: 1.5,
176
+ // num_return_sequences: 1 //must be 1 for greedy search
177
  })
178
 
179
  // let out = await pipe(inputText)