storyGeneration / requirement.txt
Thomslionel's picture
Update requirement.txt
f7b6b06 verified
raw
history blame contribute delete
146 Bytes
torch==1.10.0
transformers==4.12.3
unsloth
xformers
unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git
datasets==1.15.1
trl==0.1.0