Merge pull request #118 from NanoCode012/feat/torch-readme

Fix(readme): Fix torch missing from readme
This commit is contained in:
NanoCode012
2023-05-31 13:29:41 +09:00
committed by GitHub

View File

@@ -27,7 +27,7 @@
## Quickstart ⚡
**Requirements**: Python 3.9.
**Requirements**: Python 3.9 and Pytorch 2.0.
```bash
git clone https://github.com/OpenAccess-AI-Collective/axolotl
@@ -58,7 +58,9 @@ accelerate launch scripts/finetune.py examples/lora-openllama-3b/config.yml \
- Conda/Pip venv
1. Install python **3.9**
2. Install python dependencies with ONE of the following:
2. Install pytorch stable https://pytorch.org/get-started/locally/
3. Install python dependencies with ONE of the following:
- `pip3 install -e .` (recommended, supports QLoRA, no gptq/int4 support)
- `pip3 install -e .[gptq]` (next best if you don't need QLoRA, but want to use gptq)
- `pip3 install -e .[gptq_triton]`