unidiffuser-testing / requirements.txt
dg845's picture
Update requirements.txt
ec19bf1
# diffusers requirements
accelerate
diffusers @ git+https://github.com/dg845/diffusers@unidiffuser-pipeline
transformers
--extra-index-url https://download.pytorch.org/whl/cu118
torch
# unidiffuser requirements
# --extra-index-url https://download.pytorch.org/whl/cu116
# torch
# torchvision
# accelerate == 0.12.0
# absl-py
# ml_collections
# einops
# ftfy == 6.1.1
# transformers == 4.23.1
# clip @ git+https://github.com/openai/CLIP.git@main#egg=clip