OlivierDehaene commited on
Commit
e952e62
1 Parent(s): cb5d912

remove watermarking for open-assistant model

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -132,6 +132,8 @@ 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
  elif value == "togethercomputer/GPT-NeoXT-Chat-Base-20B":
136
  typical_p = typical_p.update(visible=False)
137
  top_p = top_p.update(value=0.25, visible=True)
@@ -246,12 +248,12 @@ with gr.Blocks(
246
  repetition_penalty = gr.Slider(
247
  minimum=0.1,
248
  maximum=3.0,
249
- value=1.01,
250
  step=0.01,
251
  interactive=True,
252
  label="Repetition Penalty",
253
  )
254
- watermark = gr.Checkbox(value=True, label="Text watermarking")
255
 
256
  model.change(
257
  lambda value: 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)
139
  top_p = top_p.update(value=0.25, visible=True)
 
248
  repetition_penalty = gr.Slider(
249
  minimum=0.1,
250
  maximum=3.0,
251
+ value=1.03,
252
  step=0.01,
253
  interactive=True,
254
  label="Repetition Penalty",
255
  )
256
+ watermark = gr.Checkbox(value=False, label="Text watermarking")
257
 
258
  model.change(
259
  lambda value: radio_on_change(