EveryDream2trainer/requirements.txt

24 lines
387 B
Plaintext

torch==2.0.1
torchvision==0.15.2
transformers==4.29.2
diffusers[torch]==0.17.1
pynvml==11.4.1
bitsandbytes==0.38.1
ftfy==6.1.1
aiohttp==3.8.4
tensorboard>=2.11.0
protobuf==3.20.1
pyre-extensions==0.0.29
xformers==0.0.20
pytorch-lightning==1.6.5
OmegaConf==2.2.3
numpy==1.23.5
dowg
lion-pytorch
compel~=1.1.3
OmegaConf==2.2.3
numpy==1.23.5
wandb
colorama
safetensors
open-flamingo==2.0.0