sarahciston commited on
Commit
bc1f72a
·
verified ·
1 Parent(s): 304ffe0

add 'Xenova/bloom-560m'

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