winglian commited on
Commit
41a4d15
·
unverified ·
1 Parent(s): 2c37bf6

update README for updated docker images (#328)

Browse files

* update README for updated docker images

* update readme from pr feedback

Files changed (1) hide show
  1. README.md +4 -5
README.md CHANGED
@@ -51,11 +51,10 @@ accelerate launch scripts/finetune.py examples/openllama-3b/lora.yml \
51
 
52
  - Docker
53
  ```bash
54
- docker run --gpus '"all"' --rm -it winglian/axolotl:main-py3.9-cu118-2.0.0
55
  ```
56
- - `winglian/axolotl-runpod:main-py3.9-cu118-2.0.0`: for runpod
57
- - `winglian/axolotl-runpod:main-py3.9-cu118-2.0.0-gptq`: for gptq
58
- - `winglian/axolotl:dev`: dev branch (not usually up to date)
59
 
60
  Or run on the current files for development:
61
 
@@ -107,7 +106,7 @@ accelerate launch scripts/finetune.py examples/openllama-3b/lora.yml \
107
 
108
  3. Install torch
109
  ```bash
110
- pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
111
  ```
112
 
113
  4. Axolotl
 
51
 
52
  - Docker
53
  ```bash
54
+ docker run --gpus '"all"' --rm -it winglian/axolotl:main-py3.10-cu118-2.0.1
55
  ```
56
+ - `winglian/axolotl-runpod:main-py3.10-cu118-2.0.1`: for runpod
57
+ - `winglian/axolotl-runpod:main-py3.9-cu118-2.0.1-gptq`: for gptq
 
58
 
59
  Or run on the current files for development:
60
 
 
106
 
107
  3. Install torch
108
  ```bash
109
+ pip3 install -U torch --index-url https://download.pytorch.org/whl/cu118
110
  ```
111
 
112
  4. Axolotl