fine_tuned_model / requirements.txt
BidhanAcharya's picture
Update requirements.txt
d8c73cf verified
raw
history blame contribute delete
187 Bytes
torch==2.0.1 # General version for both CPU and GPU environments
transformers
triton
gradio
xformers
trl<0.9.0
peft
accelerate
bitsandbytes
git+https://github.com/unslothai/unsloth.git