torch==2.7.1+cu118
transformers
gradio
accelerate
datasets
peft
trl
scikit-learn
einops
sentencepiece