joaogante HF staff commited on
Commit
da756b5
1 Parent(s): 241d9b0

attempt to fix dropdown

Browse files
Files changed (1) hide show
  1. app.py +13 -13
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=[model_selector], outputs=plot)
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"):