refactor: Update Dockerfile.tpl to use huggingface/transformers-pytorch-gpu image

This commit is contained in:
Simon Larsen 2024-06-19 13:20:34 +00:00
parent 97cc28b182
commit 85edd12c2d

View File

@ -0,0 +1,24 @@
Keep all llama models here for docker build.
# Downloading Model from Hugging Face
Please make sure you have git lfs installed before cloning the model.
```bash
git lfs install
```
```bash
# Here we are downloading the Meta-Llama-3-8B-Instruct model
git clone https://huggingface.co/meta-llama/Meta-Llama-3-8B-Instruct
```
# Larger Models
For larger models, please use the following command to download the model.
```bash
# Here we are downloading the Meta-Llama-3-8B-Instruct model
git clone https://huggingface.co/meta-llama/Meta-Llama-3-70B-Instruct
```