aixsatoshi
commited on
Commit
•
fcc5976
1
Parent(s):
6baa598
Update README.md
Browse files
README.md
CHANGED
@@ -55,9 +55,6 @@ def predict(message, history, tokens, temperature, language):
|
|
55 |
model_inputs,
|
56 |
streamer=streamer,
|
57 |
max_new_tokens=int(tokens),
|
58 |
-
do_sample=True,
|
59 |
-
top_p=0.95,
|
60 |
-
top_k=100,
|
61 |
temperature=float(temperature),
|
62 |
num_beams=1,
|
63 |
stopping_criteria=StoppingCriteriaList([stop])
|
@@ -77,7 +74,7 @@ demo = gr.ChatInterface(
|
|
77 |
title="Honyaku-7b webui",
|
78 |
description="Translate using Honyaku-7b model",
|
79 |
additional_inputs=[
|
80 |
-
gr.Slider(100,
|
81 |
gr.Slider(0.0, 1.0, value=0.3, label="Temperature"),
|
82 |
gr.Dropdown(choices=languages, value="Japanese", label="Language")
|
83 |
]
|
|
|
55 |
model_inputs,
|
56 |
streamer=streamer,
|
57 |
max_new_tokens=int(tokens),
|
|
|
|
|
|
|
58 |
temperature=float(temperature),
|
59 |
num_beams=1,
|
60 |
stopping_criteria=StoppingCriteriaList([stop])
|
|
|
74 |
title="Honyaku-7b webui",
|
75 |
description="Translate using Honyaku-7b model",
|
76 |
additional_inputs=[
|
77 |
+
gr.Slider(100, 4096, value=1000, label="Tokens"),
|
78 |
gr.Slider(0.0, 1.0, value=0.3, label="Temperature"),
|
79 |
gr.Dropdown(choices=languages, value="Japanese", label="Language")
|
80 |
]
|