mirror of
https://github.com/Stability-AI/generative-models.git
synced 2026-01-06 15:14:23 +01:00
41 lines
748 B
Plaintext
41 lines
748 B
Plaintext
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 . |