bench.py : add different large models (ggerganov#1655)

Amend different large v1,v2,v3 models to benchmark.
chidiwilliams · Dec 19, 2023 · 9286d3f · 9286d3f
1 parent 940de9d
commit 9286d3f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/extra/bench.py b/extra/bench.py
@@ -61,7 +61,9 @@ def __call__(self, parser, namespace, values, option_string=None):
     "ggml-small.bin",
     "ggml-medium.en.bin",
     "ggml-medium.bin",
-    "ggml-large.bin",
+    "ggml-large-v1.bin",
+    "ggml-large-v2.bin",
+    "ggml-large-v3.bin",
 ]