flash-diffusion / requirements.txt
clementchadebec's picture
Update requirements.txt
e23ae33 verified
raw
history blame
200 Bytes
accelerate
diffusers
invisible_watermark
--extra-index-url https://download.pytorch.org/whl/cu118
torch==2.0.1
peft >= 0.6.0
sentencepiece==0.2.0
optimum
beautifulsoup4
transformers >= 4.34.0
xformers