flash-diffusion / requirements.txt
baubin's picture
Pinned reqs
e068dde
raw
history blame contribute delete
286 Bytes
--extra-index-url https://download.pytorch.org/whl/cu121
accelerate
beautifulsoup4
ftfy
diffusers>=0.33.0
invisible_watermark
optimum
peft>=0.15.0
sentencepiece==0.2.0
spaces
torch==2.5.1
torchaudio>=2.1.0
torchvision>=0.16.0
transformers>=4.34.0,<4.49.0
xformers
xformers>=0.0.22.post7