EveryDream2trainer/requirements.txt

24 lines
381 B
Plaintext

torch==2.1.0
torchvision==0.16.0
transformers==4.35.0
diffusers[torch]==0.21.4
pynvml==11.4.1
bitsandbytes==0.41.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.22.post7
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
torchsde