joaogante HF staff commited on
Commit
b06798e
·
1 Parent(s): 74e572f

Add more Markdown

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -169,6 +169,13 @@ with demo:
169
  )
170
  with gr.Tabs():
171
  with gr.TabItem("Greedy Search"):
 
 
 
 
 
 
 
172
  model_selector = gr.Dropdown(
173
  choices=["DistilGPT2", "GPT2", "OPT-1.3B", "GPTJ-6B", "T5 Small", "T5 Base", "T5 Large", "T5 3B"],
174
  value="T5 Small",
 
169
  )
170
  with gr.Tabs():
171
  with gr.TabItem("Greedy Search"):
172
+ gr.Markdown(
173
+ """
174
+ ### Greedy Search parameters
175
+ - `max_new_tokens = 64`;
176
+ - `pad_multiple = 64` for Tensorflow XLA models. Others do not pad (input prompts between 2 and 33 tokens).
177
+ """
178
+ )
179
  model_selector = gr.Dropdown(
180
  choices=["DistilGPT2", "GPT2", "OPT-1.3B", "GPTJ-6B", "T5 Small", "T5 Base", "T5 Large", "T5 3B"],
181
  value="T5 Small",