Update app.py
Browse filestweak initial params
app.py
CHANGED
@@ -13,7 +13,7 @@ from transformers import (
|
|
13 |
)
|
14 |
|
15 |
MAX_MAX_NEW_TOKENS = 1024
|
16 |
-
DEFAULT_MAX_NEW_TOKENS =
|
17 |
MAX_INPUT_TOKEN_LENGTH = 512
|
18 |
|
19 |
DESCRIPTION = """\
|
@@ -104,14 +104,14 @@ chat_interface = gr.ChatInterface(
|
|
104 |
minimum=0.05,
|
105 |
maximum=1.0,
|
106 |
step=0.05,
|
107 |
-
value=0.
|
108 |
),
|
109 |
gr.Slider(
|
110 |
label="Top-k",
|
111 |
minimum=1,
|
112 |
maximum=1000,
|
113 |
step=1,
|
114 |
-
value=
|
115 |
),
|
116 |
gr.Slider(
|
117 |
label="Repetition penalty",
|
|
|
13 |
)
|
14 |
|
15 |
MAX_MAX_NEW_TOKENS = 1024
|
16 |
+
DEFAULT_MAX_NEW_TOKENS = 50
|
17 |
MAX_INPUT_TOKEN_LENGTH = 512
|
18 |
|
19 |
DESCRIPTION = """\
|
|
|
104 |
minimum=0.05,
|
105 |
maximum=1.0,
|
106 |
step=0.05,
|
107 |
+
value=0.5,
|
108 |
),
|
109 |
gr.Slider(
|
110 |
label="Top-k",
|
111 |
minimum=1,
|
112 |
maximum=1000,
|
113 |
step=1,
|
114 |
+
value=3,
|
115 |
),
|
116 |
gr.Slider(
|
117 |
label="Repetition penalty",
|