|
omegaconf |
|
einops |
|
fire |
|
tqdm |
|
pillow |
|
numpy |
|
webdataset>=0.2.33 |
|
--extra-index-url https://download.pytorch.org/whl/cu117 |
|
torch==1.13.1+cu117 |
|
xformers==0.0.16 |
|
torchaudio==0.13.1 |
|
torchvision==0.14.1+cu117 |
|
torchmetrics |
|
opencv-python==4.6.0.66 |
|
fairscale |
|
pytorch-lightning==1.8.5 |
|
fsspec |
|
kornia==0.6.9 |
|
matplotlib |
|
natsort |
|
tensorboardx==2.5.1 |
|
open-clip-torch |
|
chardet |
|
scipy |
|
pandas |
|
pudb |
|
pyyaml |
|
urllib3<1.27,>=1.25.4 |
|
streamlit>=0.73.1 |
|
timm |
|
tokenizers==0.12.1 |
|
torchdata==0.5.1 |
|
transformers==4.19.1 |
|
onnx<=1.12.0 |
|
triton |
|
wandb |
|
invisible-watermark |
|
-e git+https://github.com/CompVis/taming-transformers.git@master#egg=taming-transformers |
|
-e git+https://github.com/openai/CLIP.git@main#egg=clip |
|
-e git+https://github.com/Stability-AI/datapipelines.git@main#egg=sdata |
|
-e . |