diff --git a/README.md b/README.md index a81b63f1b..1be423454 100644 --- a/README.md +++ b/README.md @@ -49,15 +49,15 @@ Axolotl is a tool designed to streamline the fine-tuning of various AI models, o ## Axolotl supports -| | fp16/fp32 | lora | qlora | gptq | gptq w/ lora | gptq w/flash attn | flash attn | xformers attn | -|----------|:----------|:-----|-------|------|:-------------|-------------------|------------|---------------| -| llama | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | -| Pythia | ✅ | ✅ | ✅ | ❌ | ❓ | ❌ | ❌ | ❓ | -| cerebras | ✅ | ✅ | ✅ | ❌ | ❓ | ❌ | ❌ | ✅ | -| mpt | ✅ | ❌ | ❓ | ❌ | ❓ | ❌ | ❌ | ❓ | -| falcon | ✅ | ✅ | ✅ | ❌ | ❓ | ❌ | ❌ | ✅ | -| gpt-j | ✅ | ✅ | ✅ | ❌ | ❓ | ❌ | ❓ | ✅ | -| XGen | ✅ | ❓ | ✅ | ❓ | ❓ | ❓ | ❓ | ✅ +| | fp16/fp32 | lora | qlora | gptq | gptq w/flash attn | flash attn | xformers attn | +|----------|:----------|:-----|-------|------|-------------------|------------|---------------| +| llama | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | ✅ | +| Pythia | ✅ | ✅ | ✅ | ❌ | ❌ | ❌ | ❓ | +| cerebras | ✅ | ✅ | ✅ | ❌ | ❌ | ❌ | ❓ | +| mpt | ✅ | ❌ | ❓ | ❌ | ❌ | ❌ | ❓ | +| falcon | ✅ | ✅ | ✅ | ❌ | ❌ | ❌ | ❓ | +| gpt-j | ✅ | ✅ | ✅ | ❌ | ❌ | ❓ | ❓ | +| XGen | ✅ | ❓ | ✅ | ❓ | ❓ | ❓ | ✅ | ## Quickstart ⚡