gururise commited on
Commit
e956f7f
1 Parent(s): 417ba6e

option to reduce end fo text probability

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -251,7 +251,7 @@ iface = gr.Interface(
251
  gr.Slider(1, 256, value=40), # max_tokens
252
  gr.Slider(0.0, 1.0, value=0.8), # temperature
253
  gr.Slider(0.0, 1.0, value=0.85), # top_p
254
- gr.Slider(-99, 0.0, value=0.0, step=0.5, label="Reduce End of Text (Stop) Probability"), # end_adj
255
  gr.Textbox(lines=1, value="<|endoftext|>") # stop
256
  ],
257
  outputs=gr.Textbox(label="Generated Output", lines=25),
@@ -271,7 +271,7 @@ chatiface = gr.Interface(
271
  gr.Slider(1, 256, value=60), # max_tokens
272
  gr.Slider(0.0, 1.0, value=0.8), # temperature
273
  gr.Slider(0.0, 1.0, value=0.85), # top_p
274
- gr.Slider(-99, 0.0, value=-2, step=0.5, label="Reduce End of Text (Stop) Probability"), # end_adj
275
  ],
276
  outputs=[gr.Chatbot(label="Chat Log", color_map=(
277
  "green", "pink")), "state"],
 
251
  gr.Slider(1, 256, value=40), # max_tokens
252
  gr.Slider(0.0, 1.0, value=0.8), # temperature
253
  gr.Slider(0.0, 1.0, value=0.85), # top_p
254
+ gr.Slider(-99, 0.0, value=0.0, step=0.5, label="Reduce End of Text Probability"), # end_adj
255
  gr.Textbox(lines=1, value="<|endoftext|>") # stop
256
  ],
257
  outputs=gr.Textbox(label="Generated Output", lines=25),
 
271
  gr.Slider(1, 256, value=60), # max_tokens
272
  gr.Slider(0.0, 1.0, value=0.8), # temperature
273
  gr.Slider(0.0, 1.0, value=0.85), # top_p
274
+ gr.Slider(-99, 0.0, value=-2, step=0.5, label="Reduce End of Text Probability"), # end_adj
275
  ],
276
  outputs=[gr.Chatbot(label="Chat Log", color_map=(
277
  "green", "pink")), "state"],