EveryDream2trainer/docker/requirements.txt

20 lines
305 B
Plaintext

aiohttp==3.8.4
colorama==0.4.6
diffusers[torch]>=0.14.0
ftfy==6.1.1
ipyevents
ipywidgets
jupyter-archive
jupyterlab
ninja
omegaconf==2.2.3
piexif==1.1.3
protobuf==3.20.3
pyfakefs
pynvml==11.5.0
pyre-extensions==0.0.30
pytorch-lightning==1.9.2
tensorboard==2.12.0
transformers==4.25.1
triton>=2.0.0a2
wandb