Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ def generate(
|
|
24 |
|
25 |
generate_kwargs = dict(
|
26 |
temperature=temperature,
|
27 |
-
max_new_tokens=
|
28 |
top_p=top_p,
|
29 |
repetition_penalty=repetition_penalty,
|
30 |
do_sample=True,
|
@@ -52,15 +52,6 @@ additional_inputs=[
|
|
52 |
interactive=True,
|
53 |
info="Higher values produce more diverse outputs",
|
54 |
),
|
55 |
-
gr.Slider(
|
56 |
-
label="Max new tokens",
|
57 |
-
value=8000,
|
58 |
-
minimum=4000,
|
59 |
-
maximum=16000,
|
60 |
-
step=64,
|
61 |
-
interactive=True,
|
62 |
-
info="The maximum numbers of new tokens",
|
63 |
-
),
|
64 |
gr.Slider(
|
65 |
label="Top-p (nucleus sampling)",
|
66 |
value=0.90,
|
|
|
24 |
|
25 |
generate_kwargs = dict(
|
26 |
temperature=temperature,
|
27 |
+
max_new_tokens=4000,
|
28 |
top_p=top_p,
|
29 |
repetition_penalty=repetition_penalty,
|
30 |
do_sample=True,
|
|
|
52 |
interactive=True,
|
53 |
info="Higher values produce more diverse outputs",
|
54 |
),
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
55 |
gr.Slider(
|
56 |
label="Top-p (nucleus sampling)",
|
57 |
value=0.90,
|