story-generation-docker / requirements.txt
ThomasBlumet
return to normal config without cuda
1dd0e8c
raw
history blame
137 Bytes
#--extra-index-url https://download.pytorch.org/whl/cu118
#torch==2.0.1+cu118
torch
transformers
gradio
huggingface_hub
optimum
auto-gptq