attempt to fix dropdown
Browse files
app.py
CHANGED
|
@@ -28,9 +28,9 @@ BENCHMARK_DATA = {
|
|
| 28 |
"A100": [1125.00, 16713.63, 384.52],
|
| 29 |
},
|
| 30 |
"GPTJ-6B": {
|
| 31 |
-
"T4": [],
|
| 32 |
-
"3090": [],
|
| 33 |
-
"A100": [],
|
| 34 |
},
|
| 35 |
"T5 Small": {
|
| 36 |
"T4": [99.88, 1527.73, 18.78],
|
|
@@ -38,19 +38,19 @@ BENCHMARK_DATA = {
|
|
| 38 |
"A100": [124.91, 1642.07, 13.72],
|
| 39 |
},
|
| 40 |
"T5 Base": {
|
| 41 |
-
"T4": [],
|
| 42 |
-
"3090": [],
|
| 43 |
-
"A100": [],
|
| 44 |
},
|
| 45 |
"T5 Large": {
|
| 46 |
-
"T4": [],
|
| 47 |
-
"3090": [],
|
| 48 |
-
"A100": [],
|
| 49 |
},
|
| 50 |
"T5 3B": {
|
| 51 |
-
"T4": [],
|
| 52 |
-
"3090": [],
|
| 53 |
-
"A100": [],
|
| 54 |
},
|
| 55 |
},
|
| 56 |
"Sample": {
|
|
@@ -169,7 +169,7 @@ with demo:
|
|
| 169 |
)
|
| 170 |
plot_fn = functools.partial(get_plot, generate_type="Greedy Search")
|
| 171 |
plot = gr.Plot(value=plot_fn("T5 Small")) # Show plot when the gradio app is initialized
|
| 172 |
-
model_selector.change(fn=get_plot, inputs=
|
| 173 |
with gr.TabItem("Sample"):
|
| 174 |
gr.Button("New Tiger")
|
| 175 |
with gr.TabItem("Beam Search"):
|
|
|
|
| 28 |
"A100": [1125.00, 16713.63, 384.52],
|
| 29 |
},
|
| 30 |
"GPTJ-6B": {
|
| 31 |
+
"T4": [0, 0, 0],
|
| 32 |
+
"3090": [0, 0, 0],
|
| 33 |
+
"A100": [2664.28, 32783.09, 1440.06],
|
| 34 |
},
|
| 35 |
"T5 Small": {
|
| 36 |
"T4": [99.88, 1527.73, 18.78],
|
|
|
|
| 38 |
"A100": [124.91, 1642.07, 13.72],
|
| 39 |
},
|
| 40 |
"T5 Base": {
|
| 41 |
+
"T4": [416.56, 6095.05, 106.12],
|
| 42 |
+
"3090": [223.00, 2503.28, 46.67],
|
| 43 |
+
"A100": [550.76, 6504.11, 64.57],
|
| 44 |
},
|
| 45 |
"T5 Large": {
|
| 46 |
+
"T4": [645.05, 9587.67, 225.17],
|
| 47 |
+
"3090": [377.74, 4216.41, 97.92],
|
| 48 |
+
"A100": [944.17, 10572.43, 116.52],
|
| 49 |
},
|
| 50 |
"T5 3B": {
|
| 51 |
+
"T4": [1493.61, 13629.80, 1494.80],
|
| 52 |
+
"3090": [694.75, 6316.79, 489.33],
|
| 53 |
+
"A100": [1801.68, 16707.71, 411.93],
|
| 54 |
},
|
| 55 |
},
|
| 56 |
"Sample": {
|
|
|
|
| 169 |
)
|
| 170 |
plot_fn = functools.partial(get_plot, generate_type="Greedy Search")
|
| 171 |
plot = gr.Plot(value=plot_fn("T5 Small")) # Show plot when the gradio app is initialized
|
| 172 |
+
model_selector.change(fn=get_plot, inputs=model_selector, outputs=plot)
|
| 173 |
with gr.TabItem("Sample"):
|
| 174 |
gr.Button("New Tiger")
|
| 175 |
with gr.TabItem("Beam Search"):
|