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"):
|