diff --git a/examples/llm/slimpajama/data/download.py b/examples/llm/slimpajama/data/download.py index 813677698719..477854cb56d5 100644 --- a/examples/llm/slimpajama/data/download.py +++ b/examples/llm/slimpajama/data/download.py @@ -22,7 +22,7 @@ def download_slimpajama(include_pattern: str = "", exclude_pattern: str = ""): download_script = """ pip install "huggingface_hub[cli,hf_transfer]" -huggingface-cli download openai-community/gpt2 vocab.json merges.text --local-dir /data/bpe +huggingface-cli download openai-community/gpt2 vocab.json merges.txt --local-dir /data/bpe retry_command() { local max_retries=$1