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)
|