story-generation-docker / requirements.txt
ThomasBlumet
torch 2.0.1 with cuda 11.8
e45d465
raw
history blame
103 Bytes
transformers
torch==2.0.1+cu118
gradio
#spaces
huggingface_hub
optimum
#for running auto-gptq
auto-gptq