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 |
|