sarahciston commited on
Commit
1fd0967
·
verified ·
1 Parent(s): 2422cf3

update preprompt

Browse files
Files changed (1) hide show
  1. sketch.js +2 -2
sketch.js CHANGED
@@ -7,7 +7,7 @@ env.allowLocalModels = false;
7
 
8
  // GLOBAL VARIABLES
9
 
10
- var PREPROMPT = `Please continue each sentence after "..." and replace the "[MASK]" with your own words: `
11
  // var PREPROMPT = `Please continue the story, filling in any [MASK] with your own words:`
12
  // let PREPROMPT = `Please complete the phrase and fill in any [MASK]: `
13
  var PROMPT_INPUT = `` // a field for writing or changing a text value
@@ -54,7 +54,7 @@ async function textGenTask(pre, prompt, blanks){
54
  // const pipe = await pipeline('text-generation', MODEL) //different task type, also for text generation
55
  const pipe = await pipeline('text2text-generation', MODEL)
56
 
57
- var hyperparameters = { max_new_tokens: 200, top_k: 20, repetition_penalty: 1.5 }
58
  // setting hyperparameters
59
  // max_new_tokens: 256, top_k: 50, temperature: 0.7, do_sample: true, no_repeat_ngram_size: 2, num_return_sequences: 2 (must be 1?)
60
 
 
7
 
8
  // GLOBAL VARIABLES
9
 
10
+ var PREPROMPT = `Please continue each sentence.`/// after "..." and replace the "[MASK]" with your own words: `
11
  // var PREPROMPT = `Please continue the story, filling in any [MASK] with your own words:`
12
  // let PREPROMPT = `Please complete the phrase and fill in any [MASK]: `
13
  var PROMPT_INPUT = `` // a field for writing or changing a text value
 
54
  // const pipe = await pipeline('text-generation', MODEL) //different task type, also for text generation
55
  const pipe = await pipeline('text2text-generation', MODEL)
56
 
57
+ var hyperparameters = { max_new_tokens: 300, top_k: 30, repetition_penalty: 1.5 }
58
  // setting hyperparameters
59
  // max_new_tokens: 256, top_k: 50, temperature: 0.7, do_sample: true, no_repeat_ngram_size: 2, num_return_sequences: 2 (must be 1?)
60