sarahciston commited on
Commit
cb2c31d
1 Parent(s): 8b4c4dc

bigger model, remove preprompt

Browse files
Files changed (1) hide show
  1. sketch.js +3 -3
sketch.js CHANGED
@@ -17,11 +17,11 @@ async function textGenTask(pre, prompt){
17
  console.log('text-gen task initiated')
18
 
19
  // preprompt not working, fix later
20
- let INPUT = pre + prompt
21
- // let INPUT = prompt
22
 
23
  // PICK MODEL
24
- let MODEL = 'Xenova/LaMini-Cerebras-256M'
25
  // const = modelsList = ['Xenova/LaMini-Cerebras-256M', 'Xenova/TinyLlama-1.1B-Chat-v1.0', 'Xenova/distilgpt2', 'Xenova/bloom-560m']
26
 
27
  const pipe = await pipeline('text-generation', MODEL)
 
17
  console.log('text-gen task initiated')
18
 
19
  // preprompt not working, fix later
20
+ // let INPUT = pre + prompt
21
+ let INPUT = prompt
22
 
23
  // PICK MODEL
24
+ let MODEL = 'Xenova/LaMini-Cerebras-590M'
25
  // const = modelsList = ['Xenova/LaMini-Cerebras-256M', 'Xenova/TinyLlama-1.1B-Chat-v1.0', 'Xenova/distilgpt2', 'Xenova/bloom-560m']
26
 
27
  const pipe = await pipeline('text-generation', MODEL)