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",