Merge branch 'master' of github.com:mila-iqia/milabench

mila-iqia · Aug 23, 2024 · bdf7705 · bdf7705
2 parents a5920f9 + a3a8e41
commit bdf7705
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/llm/prepare.py b/benchmarks/llm/prepare.py
@@ -57,7 +57,7 @@ def main():
     ignore_pattern = "*.safetensors"
 
     if config.get("safetensors", False):
-        ignore_pattern = "consolidated.*.pth"
+        ignore_pattern = "*consolidated.*.pth"
 
     download_args = [
         "download",