Update app.py
Browse files
app.py
CHANGED
@@ -30,9 +30,9 @@ def respond(
|
|
30 |
|
31 |
generated_ids = model.generate(
|
32 |
**model_inputs,
|
33 |
-
max_new_tokens=max_tokens,
|
34 |
-
temperature=temperature,
|
35 |
-
top_p=top_p,
|
36 |
)
|
37 |
|
38 |
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
@@ -46,9 +46,9 @@ demo = gr.ChatInterface(
|
|
46 |
respond,
|
47 |
additional_inputs=[
|
48 |
gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
|
49 |
-
gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
|
50 |
-
gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
|
51 |
-
gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
|
52 |
],
|
53 |
)
|
54 |
|
|
|
30 |
|
31 |
generated_ids = model.generate(
|
32 |
**model_inputs,
|
33 |
+
# max_new_tokens=max_tokens,
|
34 |
+
# temperature=temperature,
|
35 |
+
# top_p=top_p,
|
36 |
)
|
37 |
|
38 |
response = tokenizer.batch_decode(generated_ids, skip_special_tokens=True)[0]
|
|
|
46 |
respond,
|
47 |
additional_inputs=[
|
48 |
gr.Textbox(value="You are a friendly Chatbot.", label="System message"),
|
49 |
+
# gr.Slider(minimum=1, maximum=2048, value=512, step=1, label="Max new tokens"),
|
50 |
+
# gr.Slider(minimum=0.1, maximum=4.0, value=0.7, step=0.1, label="Temperature"),
|
51 |
+
# gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
|
52 |
],
|
53 |
)
|
54 |
|