cahya commited on
Commit
a389ea6
·
1 Parent(s): e2c318c

rm contrastive search since it needs a lot of memory

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ with gr.Blocks(css=css) as demo:
54
  user_input = gr.inputs.Textbox(placeholder="",
55
  label="Ask me something",
56
  default="How old is the universe?")
57
- decoding_method = gr.inputs.Dropdown(["Beam Search", "Sampling", "Contrastive Search"],
58
  default="Sampling", label="Decoding Method")
59
  num_beams = gr.inputs.Slider(label="Number of beams for beam search",
60
  default=1, minimum=1, maximum=10, step=1)
 
54
  user_input = gr.inputs.Textbox(placeholder="",
55
  label="Ask me something",
56
  default="How old is the universe?")
57
+ decoding_method = gr.inputs.Dropdown(["Beam Search", "Sampling"],
58
  default="Sampling", label="Decoding Method")
59
  num_beams = gr.inputs.Slider(label="Number of beams for beam search",
60
  default=1, minimum=1, maximum=10, step=1)