21 lines
345 B
Plaintext
21 lines
345 B
Plaintext
torch==2.0.1
|
|
torchvision==0.15.2
|
|
transformers==4.29.2
|
|
diffusers[torch]==0.14.0
|
|
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 |