Update README.md for correct image tags
Browse files
README.md
CHANGED
|
@@ -50,10 +50,10 @@ accelerate launch scripts/finetune.py examples/lora-openllama-3b/config.yml \
|
|
| 50 |
|
| 51 |
- Docker
|
| 52 |
```bash
|
| 53 |
-
docker run --gpus '"all"' --rm -it winglian/axolotl:main
|
| 54 |
```
|
| 55 |
-
- `winglian/axolotl:
|
| 56 |
-
- `winglian/axolotl
|
| 57 |
|
| 58 |
Or run on the current files for development:
|
| 59 |
|
|
|
|
| 50 |
|
| 51 |
- Docker
|
| 52 |
```bash
|
| 53 |
+
docker run --gpus '"all"' --rm -it winglian/axolotl:main-py3.9-cu118-2.0.0
|
| 54 |
```
|
| 55 |
+
- `winglian/axolotl-runpod:main-py3.9-cu118-2.0.0`: for runpod
|
| 56 |
+
- `winglian/axolotl:dev`: dev branch (not usually up to date)
|
| 57 |
|
| 58 |
Or run on the current files for development:
|
| 59 |
|