I am trying to download the llama model in google colab:
Download
To download the model weights and tokenizer:

Visit the Meta Llama website.

Read and accept the license.

Once your request is approved you will receive a signed URL via email.

Install the Llama CLI: pip install llama-stack. (<-- Start Here if you have received an email already.)

Run llama model list to show the latest available models and determine the model ID you wish to download. NOTE: If you want older versions of models, run llama model list --show-all to show all the available Llama models.

Run: llama download --source meta --model-id CHOSEN_MODEL_ID

Pass the URL provided when prompted to start the download.

Remember that the links expire after 24 hours and a certain amount of downloads. You can always re-request a link if you start seeing errors such as 403: Forbidden.

Running the models
You need to install the following dependencies (in addition to the requirements.txt in the root directory of this repository) to run the models:

pip install torch fairscale fire blobfile
After installing the dependencies, you can run the example scripts (within llama_models/scripts/ sub-directory) as follows:

#!/bin/bash

CHECKPOINT_DIR=~/.llama/checkpoints/Meta-Llama3.1-8B-Instruct
PYTHONPATH=$(git rev-parse --show-toplevel) torchrun llama_models/scripts/example_chat_completion.py $CHECKPOINT_DIR
The above script should be used with an Instruct (Chat) model. For a Base model, use the script llama_models/scripts/example_text_completion.py. Note that you can use these scripts with both Llama3 and Llama3.1 series of models.

For running larger models with tensor parallelism, you should modify as:

#!/bin/bash

NGPUS=8
PYTHONPATH=$(git rev-parse --show-toplevel) torchrun \
  --nproc_per_node=$NGPUS \
  llama_models/scripts/example_chat_completion.py $CHECKPOINT_DIR \
  --model_parallel_size $NGPUS

I download it successfully :
/usr/local/lib/python3.10/dist-packages/pydantic/_internal/_fields.py:172: UserWarning: Field name "schema" in "JsonResponseFormat" shadows an attribute in parent "BaseModel"
  warnings.warn(
Please provide the signed URL for model Llama3.2-1B-Instruct you received via email after visiting https://www.llama.com/llama-downloads/ (e.g., https://llama3-1.llamameta.net/*?Policy...): https://llama3-2-lightweight.llamameta.net/*?Policy=eyJTdGF0ZW1lbnQiOlt7InVuaXF1ZV9oYXNoIjoiNTJ6MmZscTB1aWlqbHU4eHJ1bTB5anRuIiwiUmVzb3VyY2UiOiJodHRwczpcL1wvbGxhbWEzLTItbGlnaHR3ZWlnaHQubGxhbWFtZXRhLm5ldFwvKiIsIkNvbmRpdGlvbiI6eyJEYXRlTGVzc1RoYW4iOnsiQVdTOkVwb2NoVGltZSI6MTczMDUzNzE2MX19fV19&Signature=d7TPCv%7E-RHuK5rc3UmofewnwlcjbHhf84RkOPJt5UR-vvFfCj70Id8ai1AIRtcr1drDk0vFZID1dumyelgVhjOBtCQFPTDqdT9Ri0aKQLyvb9MqpxTayR1fhiPu3nEe6OnRWRcuvzN7LBnPxBIfyDY0xrb1nWtyyq%7EUd0Ua8mTl%7EuwnyVgMIGo3ssBK9sS1R5GZyg3982nPy%7EBfzB5PpqVFA-OdKXjt81IJXDbBWN%7E1kGNwPiQNjr6L0dlfme4%7ETk4OjNBnDIk6135-ZFkaTOfQ1B4InbZZ-Jk3NC0JdIjtJX7SYtP7o0yXVYEfuGh7AlbhaDTvLHcfmF9vY7JRGMg__&Key-Pair-Id=K15QRJLYKIFSLZ&Download-Request-ID=909904703979932
Downloading `checklist.chk`...
Downloading `/root/.llama/checkpoints/Llama3.2-1B-Instruct/checklist.chk`....{'Range': 'bytes=0-156'}
Progress: |██████████████████████████████████████████████████| 100.00% (0/0 MB) Speed: 0.00 MiB/s
Finished downloading `/root/.llama/checkpoints/Llama3.2-1B-Instruct/checklist.chk`....
Downloading `tokenizer.model`...
Downloading `/root/.llama/checkpoints/Llama3.2-1B-Instruct/tokenizer.model`....{'Range': 'bytes=0-2183982'}
Progress: |██████████████████████████████████████████████████| 100.00% (2/2 MB) Speed: 9.99 MiB/s
Finished downloading `/root/.llama/checkpoints/Llama3.2-1B-Instruct/tokenizer.model`....
Downloading `params.json`...
Downloading `/root/.llama/checkpoints/Llama3.2-1B-Instruct/params.json`....{'Range': 'bytes=0-220'}
Progress: |██████████████████████████████████████████████████| 100.00% (0/0 MB) Speed: 0.00 MiB/s
Finished downloading `/root/.llama/checkpoints/Llama3.2-1B-Instruct/params.json`....
Downloading `consolidated.00.pth`...
Downloading `/root/.llama/checkpoints/Llama3.2-1B-Instruct/consolidated.00.pth`....{'Range': 'bytes=0-2471677246'}
Progress: |██████████████████████████████████████████████████| 100.00% (2357/2357 MB) Speed: 34.85 MiB/s
Finished downloading `/root/.llama/checkpoints/Llama3.2-1B-Instruct/consolidated.00.pth`....

Successfully downloaded model to /root/.llama/checkpoints/Llama3.2-1B-Instruct

MD5 Checksums are at: /root/.llama/checkpoints/Llama3.2-1B-Instruct/checklist.chk
Write the code to save the model to my google drive.