unsloth
torch>=2.5.1
transformers>=4.45
trl
peft
bitsandbytes
datasets
accelerate
gradio
pandas
huggingface_hub
pyyaml

[dev]
pytest
black
isort

[gpu]
torch==2.5.1+cu121
bitsandbytes
nvidia-cuda-runtime-cu12
nvidia-cuda-cupti-cu12
nvidia-cudnn-cu12
nvidia-cublas-cu12
nvidia-cuda-nvrtc-cu12
nvidia-curand-cu12
nvidia-cusolver-cu12
nvidia-cusparse-cu12
nvidia-nccl-cu12
nvidia-nvtx-cu12
