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