zongheng commited on
Commit
b33c1d5
1 Parent(s): 0c2a630

Docs: add instructions to 1-click launching on public clouds (#862)

Browse files
Files changed (1) hide show
  1. README.md +24 -0
README.md CHANGED
@@ -25,8 +25,10 @@ Features:
25
  - [Installation](#installation)
26
  - [Docker](#docker)
27
  - [Conda/Pip venv](#condapip-venv)
 
28
  - [LambdaLabs](#lambdalabs)
29
  - [Windows](#windows)
 
30
  - [Dataset](#dataset)
31
  - [How to Add Custom Prompts](#how-to-add-custom-prompts)
32
  - [How to Use Custom Pretokenized Dataset](#how-to-use-your-custom-pretokenized-dataset)
@@ -204,6 +206,28 @@ Use `winglian/axolotl-runpod:main-latest` or use this [direct link](https://runp
204
  #### Windows
205
  Please use WSL or Docker!
206
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
207
  ### Dataset
208
 
209
  Axolotl supports a variety of dataset formats. Below are some of the formats you can use.
 
25
  - [Installation](#installation)
26
  - [Docker](#docker)
27
  - [Conda/Pip venv](#condapip-venv)
28
+ - [Runpod](#runpod)
29
  - [LambdaLabs](#lambdalabs)
30
  - [Windows](#windows)
31
+ - [Launching on public clouds via SkyPilot](#launching-on-public-clouds-via-skypilot)
32
  - [Dataset](#dataset)
33
  - [How to Add Custom Prompts](#how-to-add-custom-prompts)
34
  - [How to Use Custom Pretokenized Dataset](#how-to-use-your-custom-pretokenized-dataset)
 
206
  #### Windows
207
  Please use WSL or Docker!
208
 
209
+
210
+ #### Launching on public clouds via SkyPilot
211
+ To launch on GPU instances (both on-demand and spot instances) on 7+ clouds (GCP, AWS, Azure, OCI, and more), you can use [SkyPilot](https://skypilot.readthedocs.io/en/latest/index.html):
212
+ ```bash
213
+ pip install "skypilot-nightly[gcp,aws,azure,oci,lambda,kubernetes,ibm,scp]" # choose your clouds
214
+ sky check
215
+ ```
216
+ Get the [example YAMLs](https://github.com/skypilot-org/skypilot/tree/master/llm/axolotl) of using Axolotl to finetune `mistralai/Mistral-7B-v0.1`:
217
+ ```
218
+ git clone https://github.com/skypilot-org/skypilot.git
219
+ cd skypilot/llm/axolotl
220
+ ```
221
+ Use one command to launch:
222
+ ```bash
223
+ # On-demand
224
+ HF_TOKEN=xx sky launch axolotl.yaml --env HF_TOKEN
225
+
226
+ # Managed spot (auto-recovery on preemption)
227
+ HF_TOKEN=xx BUCKET=<unique-name> sky spot launch axolotl-spot.yaml --env HF_TOKEN --env BUCKET
228
+ ```
229
+
230
+
231
  ### Dataset
232
 
233
  Axolotl supports a variety of dataset formats. Below are some of the formats you can use.