Spaces:
Running
Running
acumplid
commited on
Commit
·
fb329f5
1
Parent(s):
359b142
Modified params and top_p
Browse files
app.py
CHANGED
|
@@ -45,13 +45,12 @@ def submit_input(input_, num_chunks, max_new_tokens, repetition_penalty, top_k,
|
|
| 45 |
|
| 46 |
model_parameters = {
|
| 47 |
"NUM_CHUNKS": num_chunks,
|
| 48 |
-
"
|
| 49 |
-
"
|
| 50 |
-
"
|
| 51 |
-
"
|
| 52 |
-
"
|
| 53 |
-
"
|
| 54 |
-
"TEMPERATURE": temperature
|
| 55 |
}
|
| 56 |
|
| 57 |
output, context, source = generate(input_, model_parameters)
|
|
@@ -152,8 +151,8 @@ def gradio_app():
|
|
| 152 |
)
|
| 153 |
top_p = Slider(
|
| 154 |
minimum=0.01,
|
| 155 |
-
maximum=
|
| 156 |
-
value=
|
| 157 |
label="Top p"
|
| 158 |
)
|
| 159 |
do_sample = Checkbox(
|
|
|
|
| 45 |
|
| 46 |
model_parameters = {
|
| 47 |
"NUM_CHUNKS": num_chunks,
|
| 48 |
+
"max_new_tokens": max_new_tokens,
|
| 49 |
+
"repetition_penalty": repetition_penalty,
|
| 50 |
+
"top_k": top_k,
|
| 51 |
+
"top_p": top_p,
|
| 52 |
+
"do_sample": do_sample,
|
| 53 |
+
"temperature": temperature
|
|
|
|
| 54 |
}
|
| 55 |
|
| 56 |
output, context, source = generate(input_, model_parameters)
|
|
|
|
| 151 |
)
|
| 152 |
top_p = Slider(
|
| 153 |
minimum=0.01,
|
| 154 |
+
maximum=0.99,
|
| 155 |
+
value=0.99,
|
| 156 |
label="Top p"
|
| 157 |
)
|
| 158 |
do_sample = Checkbox(
|