story-generation-docker / requirements.txt
ThomasBlumet
update including pip and cuda version
69a1d98
raw
history blame
102 Bytes
transformers
torch=2.0.1+cu113
gradio
#spaces
huggingface_hub
optimum
#for running auto-gptq
auto-gptq