sarahciston commited on
Commit
40be2d7
·
verified ·
1 Parent(s): bd2d4fa

can't do num_return_sequences on greedy search

Browse files
Files changed (1) hide show
  1. sketch.js +1 -1
sketch.js CHANGED
@@ -27,7 +27,7 @@ async function textGenTask(pre, prompt){
27
  const pipe = await pipeline('text-generation', MODEL)
28
 
29
  // RUN INPUT THROUGH MODEL,
30
- var out = await pipe(INPUT, { max_new_tokens: 60, top_k: 90, num_return_sequences: 3 })
31
  // setting hyperparameters
32
  // max_new_tokens: 256, top_k: 50, temperature: 0.7, do_sample: true,
33
 
 
27
  const pipe = await pipeline('text-generation', MODEL)
28
 
29
  // RUN INPUT THROUGH MODEL,
30
+ var out = await pipe(INPUT, { max_new_tokens: 60, top_k: 90 })
31
  // setting hyperparameters
32
  // max_new_tokens: 256, top_k: 50, temperature: 0.7, do_sample: true,
33