tocmo0nlord/axolotl
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

update README for updated docker images (#328)

Browse Source 
* update README for updated docker images

* update readme from pr feedback
This commit is contained in:
Wing Lian
2023-07-28 16:50:03 -04:00
committed by GitHub
parent 2c37bf6c21
commit 41a4d15d43
1 changed files with 4 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
README.md
View File

@@ -51,11 +51,10 @@ accelerate launch scripts/finetune.py examples/openllama-3b/lora.yml \
- Docker - Docker
```bash ```bash
docker run --gpus '"all"' --rm -it winglian/axolotl:main-py3.9-cu118-2.0.0 docker run --gpus '"all"' --rm -it winglian/axolotl:main-py3.10-cu118-2.0.1
``` ```
- `winglian/axolotl-runpod:main-py3.9-cu118-2.0.0`: for runpod - `winglian/axolotl-runpod:main-py3.10-cu118-2.0.1`: for runpod
- `winglian/axolotl-runpod:main-py3.9-cu118-2.0.0-gptq`: for gptq - `winglian/axolotl-runpod:main-py3.9-cu118-2.0.1-gptq`: for gptq
- `winglian/axolotl:dev`: dev branch (not usually up to date)
Or run on the current files for development: Or run on the current files for development:
@@ -107,7 +106,7 @@ accelerate launch scripts/finetune.py examples/openllama-3b/lora.yml \
3. Install torch 3. Install torch
```bash ```bash
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118 pip3 install -U torch --index-url https://download.pytorch.org/whl/cu118
``` ```
4. Axolotl 4. Axolotl