training-gpt-2 / requirements.txt
vakodiya's picture
Added models and dataset for training
2032ac8
raw
history blame
103 Bytes
streamlit
transformers==4.41.2
torch==2.3.1
datasets==2.20.0
huggingface_hub==0.23.2
accelerate==0.32.1