diff --git a/bench/exo_bench.py b/bench/exo_bench.py index e60d0872..001f5c56 100644 --- a/bench/exo_bench.py +++ b/bench/exo_bench.py @@ -75,7 +75,7 @@ def load_tokenizer_for_bench(model_id: str) -> Any: model_path = Path( snapshot_download( model_id, - allow_patterns=["*.json", "*.py", "*.tiktoken"], + allow_patterns=["*.json", "*.py", "*.tiktoken", "*.model"], ) )