Nanobit commited on
Commit
9aab0b8
1 Parent(s): 857a80b

Update Docker instructions

Browse files
Files changed (1) hide show
  1. README.md +7 -3
README.md CHANGED
@@ -20,8 +20,12 @@ Go ahead and axolotl questions!!
20
 
21
  - Docker
22
  ```bash
23
- docker pull winglian/axolotl
24
  ```
 
 
 
 
25
 
26
  - Conda/Pip venv
27
  1. Install python **3.9**
@@ -33,7 +37,7 @@ Go ahead and axolotl questions!!
33
 
34
  ### Dataset
35
 
36
- Have a dataset in one of the following format (JSONL recommended):
37
 
38
  - `alpaca`: instruction; input(optional)
39
  ```json
@@ -64,7 +68,7 @@ Have a dataset in one of the following format (JSONL recommended):
64
  {"text": "..."}
65
  ```
66
 
67
- Optionally Download some datasets, see [data/README.md](data/README.md)
68
 
69
  ### Config
70
 
 
20
 
21
  - Docker
22
  ```bash
23
+ docker run --gpus '"all"' --rm -it winglian/axolotl:main
24
  ```
25
+ - `winglian/axolotl:dev`: dev branch
26
+ - `winglian/axolotl-runpod:main`: for runpod
27
+ - `--gpus '"device=0"'`: to select one gpu
28
+ - `-v $PWD:/workspace/axolotl`: to mount current path for dev
29
 
30
  - Conda/Pip venv
31
  1. Install python **3.9**
 
37
 
38
  ### Dataset
39
 
40
+ Have dataset(s) in one of the following format (JSONL recommended):
41
 
42
  - `alpaca`: instruction; input(optional)
43
  ```json
 
68
  {"text": "..."}
69
  ```
70
 
71
+ Optionally, download some datasets, see [data/README.md](data/README.md)
72
 
73
  ### Config
74