2023-11-02 19:54:29 -06:00
|
|
|
torch==2.1.0
|
|
|
|
|
torchvision==0.16.0
|
2024-03-13 16:38:30 -06:00
|
|
|
transformers>=4.38.2
|
2023-11-02 19:54:29 -06:00
|
|
|
diffusers[torch]==0.21.4
|
2024-03-13 16:38:30 -06:00
|
|
|
peft>=0.9.0
|
2023-05-30 20:15:02 -06:00
|
|
|
pynvml==11.4.1
|
2023-08-31 10:46:46 -06:00
|
|
|
bitsandbytes==0.41.1
|
2023-05-30 20:15:02 -06:00
|
|
|
ftfy==6.1.1
|
|
|
|
|
aiohttp==3.8.4
|
|
|
|
|
tensorboard>=2.11.0
|
|
|
|
|
protobuf==3.20.1
|
|
|
|
|
pyre-extensions==0.0.29
|
2023-11-17 10:31:25 -07:00
|
|
|
xformers==0.0.22.post7
|
2023-05-30 20:15:02 -06:00
|
|
|
pytorch-lightning==1.6.5
|
|
|
|
|
OmegaConf==2.2.3
|
|
|
|
|
numpy==1.23.5
|
2023-06-08 09:39:46 -06:00
|
|
|
dowg
|
2023-05-30 20:15:02 -06:00
|
|
|
lion-pytorch
|
|
|
|
|
compel~=1.1.3
|
|
|
|
|
OmegaConf==2.2.3
|
|
|
|
|
numpy==1.23.5
|
2023-06-12 22:06:30 -06:00
|
|
|
wandb
|
|
|
|
|
colorama
|
2023-06-29 16:12:52 -06:00
|
|
|
safetensors
|
2024-06-08 23:25:23 -06:00
|
|
|
torchsde
|
|
|
|
|
tiktoken
|
