Update app.py
Browse files
app.py
CHANGED
@@ -57,9 +57,9 @@ def respond(
|
|
57 |
response_output = generator(
|
58 |
prompt,
|
59 |
generation_config=generator_conf,
|
60 |
-
max_new_tokens=
|
61 |
do_sample=True,
|
62 |
-
top_p=
|
63 |
repetition_penalty=1.2
|
64 |
)
|
65 |
|
@@ -99,12 +99,12 @@ demo = gr.ChatInterface(
|
|
99 |
respond,
|
100 |
additional_inputs=[
|
101 |
gr.Textbox(value="You are a friendly Chatbot." + global_error, label="System message"),
|
102 |
-
gr.Slider(minimum=1, maximum=2048, value=
|
103 |
-
gr.Slider(minimum=0.1, maximum=4.0, value=0
|
104 |
gr.Slider(
|
105 |
minimum=0.1,
|
106 |
maximum=1.0,
|
107 |
-
value=0.
|
108 |
step=0.05,
|
109 |
label="Top-p (nucleus sampling)",
|
110 |
),
|
|
|
57 |
response_output = generator(
|
58 |
prompt,
|
59 |
generation_config=generator_conf,
|
60 |
+
max_new_tokens=max_tokens,
|
61 |
do_sample=True,
|
62 |
+
top_p=top_p,
|
63 |
repetition_penalty=1.2
|
64 |
)
|
65 |
|
|
|
99 |
respond,
|
100 |
additional_inputs=[
|
101 |
gr.Textbox(value="You are a friendly Chatbot." + global_error, label="System message"),
|
102 |
+
gr.Slider(minimum=1, maximum=2048, value=64, step=1, label="Max new tokens"),
|
103 |
+
# gr.Slider(minimum=0.1, maximum=4.0, value=1.0, step=0.1, label="Temperature"),
|
104 |
gr.Slider(
|
105 |
minimum=0.1,
|
106 |
maximum=1.0,
|
107 |
+
value=0.25,
|
108 |
step=0.05,
|
109 |
label="Top-p (nucleus sampling)",
|
110 |
),
|