From 107b67b852badb4d269e8100e76b544ac220d4aa Mon Sep 17 00:00:00 2001 From: Oliver Kunc <36070570+OliverKunc@users.noreply.github.com> Date: Tue, 29 Oct 2024 15:13:50 +0100 Subject: [PATCH] Hardware requirements (#1997) [skip ci] * Hardware requirements https://github.com/axolotl-ai-cloud/axolotl/issues/1992 * Update README.md --------- Co-authored-by: Wing Lian --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 21b954a56..c12aa3bba 100644 --- a/README.md +++ b/README.md @@ -121,7 +121,7 @@ Features: Get started with Axolotl in just a few steps! This quickstart guide will walk you through setting up and running a basic fine-tuning task. -**Requirements**: Python >=3.10 and Pytorch >=2.1.1. +**Requirements**: Nvidia GPU (Ampere architecture or newer for `bf16` and Flash Attention), Python >=3.10 and PyTorch >=2.3.1. ```bash git clone https://github.com/axolotl-ai-cloud/axolotl