update README for updated docker images (#328)
Browse files* update README for updated docker images
* update readme from pr feedback
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.
|
55 |
```
|
56 |
-
- `winglian/axolotl-runpod:main-py3.
|
57 |
-
- `winglian/axolotl-runpod:main-py3.9-cu118-2.0.
|
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
|
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
|