add Beam Search data
Browse files
app.py
CHANGED
|
@@ -97,44 +97,44 @@ BENCHMARK_DATA = {
|
|
| 97 |
},
|
| 98 |
"Beam Search": {
|
| 99 |
"DistilGPT2": {
|
| 100 |
-
"T4": [],
|
| 101 |
-
"3090": [],
|
| 102 |
-
"A100": [],
|
| 103 |
},
|
| 104 |
"GPT2": {
|
| 105 |
-
"T4": [],
|
| 106 |
-
"3090": [],
|
| 107 |
-
"A100": [],
|
| 108 |
},
|
| 109 |
"OPT-1.3B": {
|
| 110 |
-
"T4": [],
|
| 111 |
-
"3090": [],
|
| 112 |
-
"A100": [],
|
| 113 |
},
|
| 114 |
"GPTJ-6B": {
|
| 115 |
-
"T4": [],
|
| 116 |
-
"3090": [],
|
| 117 |
-
"A100": [],
|
| 118 |
},
|
| 119 |
"T5 Small": {
|
| 120 |
-
"T4": [],
|
| 121 |
-
"3090": [],
|
| 122 |
-
"A100": [],
|
| 123 |
},
|
| 124 |
"T5 Base": {
|
| 125 |
-
"T4": [],
|
| 126 |
-
"3090": [],
|
| 127 |
-
"A100": [],
|
| 128 |
},
|
| 129 |
"T5 Large": {
|
| 130 |
-
"T4": [],
|
| 131 |
-
"3090": [],
|
| 132 |
-
"A100": [],
|
| 133 |
},
|
| 134 |
"T5 3B": {
|
| 135 |
-
"T4": [],
|
| 136 |
-
"3090": [],
|
| 137 |
-
"A100": [],
|
| 138 |
},
|
| 139 |
},
|
| 140 |
}
|
|
|
|
| 97 |
},
|
| 98 |
"Beam Search": {
|
| 99 |
"DistilGPT2": {
|
| 100 |
+
"T4": [2407.89, 19442.60, 3313.92],
|
| 101 |
+
"3090": [998.52, 8286.03, 900.28],
|
| 102 |
+
"A100": [2237.41, 21771.40, 760.47],
|
| 103 |
},
|
| 104 |
"GPT2": {
|
| 105 |
+
"T4": [3767.43, 34813.93, 5559.42],
|
| 106 |
+
"3090": [1633.04, 14606.93, 1533.55],
|
| 107 |
+
"A100": [3705.43, 34586.23, 1295.87],
|
| 108 |
},
|
| 109 |
"OPT-1.3B": {
|
| 110 |
+
"T4": [16649.82, 78500.33, 21894.31],
|
| 111 |
+
"3090": [508518, 32822.81, 5762.46],
|
| 112 |
+
"A100": [5967.32, 78334.56, 4096.38],
|
| 113 |
},
|
| 114 |
"GPTJ-6B": {
|
| 115 |
+
"T4": [0, 0, 0],
|
| 116 |
+
"3090": [0, 0, 0],
|
| 117 |
+
"A100": [15119.10, 134000.40, 10214.17],
|
| 118 |
},
|
| 119 |
"T5 Small": {
|
| 120 |
+
"T4": [283.64, 25089.12, 1391.66],
|
| 121 |
+
"3090": [137.38, 10680.28, 486.96],
|
| 122 |
+
"A100": [329.28, 24747.38, 513.99],
|
| 123 |
},
|
| 124 |
"T5 Base": {
|
| 125 |
+
"T4": [1383.21, 44809.14, 3920.40],
|
| 126 |
+
"3090": [723.11, 18657.48, 1258.60],
|
| 127 |
+
"A100": [2360.85, 45085.07, 1107.58],
|
| 128 |
},
|
| 129 |
"T5 Large": {
|
| 130 |
+
"T4": [1663.50, 81902.41, 9551.29],
|
| 131 |
+
"3090": [922.53, 35524.30, 2838.86],
|
| 132 |
+
"A100": [2168.22, 86890.00, 2373.04],
|
| 133 |
},
|
| 134 |
"T5 3B": {
|
| 135 |
+
"T4": [0, 0, 0],
|
| 136 |
+
"3090": [1521.05, 35337.30, 8282.09],
|
| 137 |
+
"A100": [3162.54, 88453.65, 5585.20],
|
| 138 |
},
|
| 139 |
},
|
| 140 |
}
|