fluffyvision / requirements.txt
j4y1337's picture
gpu preparation
2e31314 verified
--extra-index-url https://download.pytorch.org/whl/cu117
Pillow>=8.3.2
deep_translator>=1.4.4
numpy>=1.25.0
transformers>=4.38.2
accelerate>=0.28.0
torch==2.0.1+cu117
torchvision==0.15.2+cu117
torchaudio==2.0.2
bitsandbytes-cuda117>=0.26.0.post2
bitsandbytes-windows>=0.37.5
bitsandbytes>=0.43.0