Docs: add instructions to 1-click launching on public clouds (#862)
Browse files
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.
|