EveryDream2trainer/requirements-test.txt

22 lines
314 B
Plaintext

diffusers[torch]>=0.14.0
ninja
numpy
omegaconf==2.2.3
protobuf==3.20.3
pyre-extensions==0.0.23
pytorch-lightning==1.9.2
transformers==4.27.1
aiohttp==3.8.4
bitsandbytes==0.37.2
colorama==0.4.6
compel~=1.1.3
ftfy==6.1.1
lion-pytorch
piexif==1.1.3
pyfakefs
pynvml==11.5.0
tensorboard==2.12.0
torch
torchvision
wandb