requirements.txt

pip==24.1.2
--find-links https://download.pytorch.org/whl/cu121
torch==2.2.0
triton

git+https://github.com/unslothai/unsloth.git#egg=unsloth[cu121-ampere-torch220]

transformers
accelerate
bitsandbytes
einops
xformers

flash-attn --no-build-isolation


numpy
packaging