Spaces:
Running
Running
sarahciston
commited on
Commit
•
a244abf
1
Parent(s):
468bd4a
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)
|