OlivierDehaene commited on
Commit
35539cb
1 Parent(s): e952e62

remove repetition penalty for open-assistant

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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(value=1.03, visible=True)
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