custom_gpt2 / requirements.txt
Gbssreejith's picture
Update requirements.py
bbd227d verified
raw
history blame
174 Bytes
torch==2.0.0+cu118
torchvision==0.15.0+cu118
torchaudio==2.0.0+cu118
-f https://download.pytorch.org/whl/torch_stable.html
transformers
gradio
pandas
scikit-learn
numpy
nltk