gpt2FineTuned / requirements.txt
shrivasc's picture
Update requirements.txt
bb8ba55 verified
raw
history blame contribute delete
70 Bytes
transformers==4.31.0
torch==2.0.1 # or torch>=2.0.0
accelerate
gradio