Spaces:
Running
Running
new prompt new hyperparams
Browse files
sketch.js
CHANGED
|
@@ -6,7 +6,7 @@ import { pipeline, env } from 'https://cdn.jsdelivr.net/npm/@xenova/transformers
|
|
| 6 |
env.allowLocalModels = false;
|
| 7 |
|
| 8 |
// GLOBAL VARIABLES
|
| 9 |
-
let PROMPT_INPUT = `
|
| 10 |
let pField
|
| 11 |
|
| 12 |
// RUN TEXT-GEN MODEL
|
|
@@ -20,9 +20,10 @@ async function textGenTask(input){
|
|
| 20 |
temperature: 2,
|
| 21 |
max_new_tokens: 200,
|
| 22 |
repetition_penalty: 2,
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
|
|
|
| 26 |
})
|
| 27 |
|
| 28 |
console.log(await out)
|
|
|
|
| 6 |
env.allowLocalModels = false;
|
| 7 |
|
| 8 |
// GLOBAL VARIABLES
|
| 9 |
+
let PROMPT_INPUT = `The woman has a job as a...` // a field for writing or changing a text value
|
| 10 |
let pField
|
| 11 |
|
| 12 |
// RUN TEXT-GEN MODEL
|
|
|
|
| 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)
|