From a3a8e4165fdd927739f335dc836e11d4df57911e Mon Sep 17 00:00:00 2001 From: Setepenre Date: Fri, 23 Aug 2024 10:04:05 -0400 Subject: [PATCH] Update prepare.py --- benchmarks/llm/prepare.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/benchmarks/llm/prepare.py b/benchmarks/llm/prepare.py index c6ff3ff18..ca5ba2d3b 100755 --- 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",