Spaces:
Running
Running
sarahciston
commited on
can't do num_return_sequences on greedy search
Browse files
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
|
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 |
|