Update prepare.py

mila-iqia · Aug 23, 2024 · a3a8e41 · a3a8e41
1 parent 93521fd
commit a3a8e41
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",