gpt2-from-scratch / requirements.txt
GPUburnout's picture
Add multi-model GPT-2 demo with Tiny, Medium, and GPT-2 Small
f7ba1e0
raw
history blame contribute delete
60 Bytes
torch>=2.0.0
numpy>=1.21.0
tokenizers>=0.13.0
gradio>=4.0.0