no, column is better
Browse files
app.py
CHANGED
@@ -186,27 +186,28 @@ with demo:
|
|
186 |
with gr.Tabs():
|
187 |
with gr.TabItem("Greedy Search"):
|
188 |
plot_fn = functools.partial(get_plot, generate_type="Greedy Search")
|
189 |
-
gr.Markdown(
|
190 |
-
"""
|
191 |
-
### Greedy Search benchmark parameters
|
192 |
-
- `max_new_tokens = 64`;
|
193 |
-
- `pad_to_multiple_of = 64` for Tensorflow XLA models. Others do not pad (input prompts between 2 and 33 tokens).
|
194 |
-
"""
|
195 |
-
)
|
196 |
with gr.Row():
|
197 |
-
|
198 |
-
|
199 |
-
|
200 |
-
|
201 |
-
|
202 |
-
|
203 |
-
|
204 |
-
|
205 |
-
|
206 |
-
|
207 |
-
|
208 |
-
|
209 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
210 |
model_selector.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
211 |
eager_enabler.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
212 |
with gr.TabItem("Sample"):
|
|
|
186 |
with gr.Tabs():
|
187 |
with gr.TabItem("Greedy Search"):
|
188 |
plot_fn = functools.partial(get_plot, generate_type="Greedy Search")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
189 |
with gr.Row():
|
190 |
+
with gr.Column():
|
191 |
+
model_selector = gr.Dropdown(
|
192 |
+
choices=["DistilGPT2", "GPT2", "OPT-1.3B", "GPTJ-6B", "T5 Small", "T5 Base", "T5 Large", "T5 3B"],
|
193 |
+
value="T5 Small",
|
194 |
+
label="Model",
|
195 |
+
interactive=True,
|
196 |
+
)
|
197 |
+
eager_enabler = gr.Radio(
|
198 |
+
["Yes", "No"],
|
199 |
+
value="Yes",
|
200 |
+
label="Plot TF Eager Execution?",
|
201 |
+
interactive=True
|
202 |
+
)
|
203 |
+
gr.Markdown(
|
204 |
+
"""
|
205 |
+
### Greedy Search benchmark parameters
|
206 |
+
- `max_new_tokens = 64`;
|
207 |
+
- `pad_to_multiple_of = 64` for Tensorflow XLA models. Others do not pad (input prompts between 2 and 33 tokens).
|
208 |
+
"""
|
209 |
+
)
|
210 |
+
plot = gr.Image(value=plot_fn("T5 Small", "Yes")) # Show plot when the gradio app is initialized
|
211 |
model_selector.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
212 |
eager_enabler.change(fn=plot_fn, inputs=[model_selector, eager_enabler], outputs=plot)
|
213 |
with gr.TabItem("Sample"):
|