Spaces:
Running
Running
remove some hyperparams
Browse files
sketch.js
CHANGED
|
@@ -17,13 +17,13 @@ async function textGenTask(input){
|
|
| 17 |
const pipe = await pipeline('text-generation', 'Xenova/LaMini-Cerebras-256M')
|
| 18 |
|
| 19 |
var out = await pipe(input, {
|
| 20 |
-
temperature: 2,
|
| 21 |
max_new_tokens: 200,
|
| 22 |
-
repetition_penalty: 2
|
| 23 |
-
min_new_tokens: 100
|
| 24 |
-
no_repeat_ngram_size: 2,
|
| 25 |
-
num_beams: 2,
|
| 26 |
-
num_return_sequences: 2
|
| 27 |
})
|
| 28 |
|
| 29 |
console.log(await out)
|
|
|
|
| 17 |
const pipe = await pipeline('text-generation', 'Xenova/LaMini-Cerebras-256M')
|
| 18 |
|
| 19 |
var out = await pipe(input, {
|
| 20 |
+
// temperature: 2,
|
| 21 |
max_new_tokens: 200,
|
| 22 |
+
repetition_penalty: 2
|
| 23 |
+
min_new_tokens: 100
|
| 24 |
+
// no_repeat_ngram_size: 2,
|
| 25 |
+
// num_beams: 2,
|
| 26 |
+
// num_return_sequences: 2
|
| 27 |
})
|
| 28 |
|
| 29 |
console.log(await out)
|