Spaces:
Runtime error
Runtime error
OlivierDehaene
commited on
Commit
·
35539cb
1
Parent(s):
e952e62
remove repetition penalty for open-assistant
Browse files
app.py
CHANGED
|
@@ -73,7 +73,6 @@ def predict(
|
|
| 73 |
iterator = client.generate_stream(
|
| 74 |
total_inputs,
|
| 75 |
typical_p=typical_p,
|
| 76 |
-
repetition_penalty=repetition_penalty,
|
| 77 |
watermark=watermark,
|
| 78 |
max_new_tokens=500,
|
| 79 |
)
|
|
@@ -132,7 +131,7 @@ def radio_on_change(
|
|
| 132 |
top_k = top_k.update(visible=False)
|
| 133 |
temperature = temperature.update(visible=False)
|
| 134 |
disclaimer = disclaimer.update(visible=False)
|
| 135 |
-
repetition_penalty = repetition_penalty.update(
|
| 136 |
watermark = watermark.update(False)
|
| 137 |
elif value == "togethercomputer/GPT-NeoXT-Chat-Base-20B":
|
| 138 |
typical_p = typical_p.update(visible=False)
|
|
@@ -147,7 +146,7 @@ def radio_on_change(
|
|
| 147 |
top_p = top_p.update(value=0.95, visible=True)
|
| 148 |
top_k = top_k.update(value=4, visible=True)
|
| 149 |
temperature = temperature.update(value=0.5, visible=True)
|
| 150 |
-
repetition_penalty = repetition_penalty.update(value=1.03)
|
| 151 |
watermark = watermark.update(True)
|
| 152 |
disclaimer = disclaimer.update(visible=False)
|
| 153 |
return (
|
|
@@ -252,6 +251,7 @@ with gr.Blocks(
|
|
| 252 |
step=0.01,
|
| 253 |
interactive=True,
|
| 254 |
label="Repetition Penalty",
|
|
|
|
| 255 |
)
|
| 256 |
watermark = gr.Checkbox(value=False, label="Text watermarking")
|
| 257 |
|
|
|
|
| 73 |
iterator = client.generate_stream(
|
| 74 |
total_inputs,
|
| 75 |
typical_p=typical_p,
|
|
|
|
| 76 |
watermark=watermark,
|
| 77 |
max_new_tokens=500,
|
| 78 |
)
|
|
|
|
| 131 |
top_k = top_k.update(visible=False)
|
| 132 |
temperature = temperature.update(visible=False)
|
| 133 |
disclaimer = disclaimer.update(visible=False)
|
| 134 |
+
repetition_penalty = repetition_penalty.update(visible=False)
|
| 135 |
watermark = watermark.update(False)
|
| 136 |
elif value == "togethercomputer/GPT-NeoXT-Chat-Base-20B":
|
| 137 |
typical_p = typical_p.update(visible=False)
|
|
|
|
| 146 |
top_p = top_p.update(value=0.95, visible=True)
|
| 147 |
top_k = top_k.update(value=4, visible=True)
|
| 148 |
temperature = temperature.update(value=0.5, visible=True)
|
| 149 |
+
repetition_penalty = repetition_penalty.update(value=1.03, visible=True)
|
| 150 |
watermark = watermark.update(True)
|
| 151 |
disclaimer = disclaimer.update(visible=False)
|
| 152 |
return (
|
|
|
|
| 251 |
step=0.01,
|
| 252 |
interactive=True,
|
| 253 |
label="Repetition Penalty",
|
| 254 |
+
visible=False,
|
| 255 |
)
|
| 256 |
watermark = gr.Checkbox(value=False, label="Text watermarking")
|
| 257 |
|