gpt2-chatbot / requirements.txt
greyfoss's picture
Update requirements.txt
a264b43 verified
raw
history blame
137 Bytes
torch
numpy==1.25.2
transformers==4.30.2
datasets==2.14.4
sentencepiece==0.1.99
optimum[exporters]
onnxruntime
accelerate
huggingface_hub