mirror of
https://github.com/Stability-AI/stablediffusion.git
synced 2024-12-22 15:44:58 +00:00
30 lines
686 B
YAML
30 lines
686 B
YAML
name: ldm
|
|
channels:
|
|
- pytorch
|
|
- defaults
|
|
dependencies:
|
|
# Changed to Python 3.9 to make use of precompiled binaries for xformers
|
|
- python=3.9
|
|
- pip=20.3
|
|
- cudatoolkit=11.3
|
|
- pytorch=1.12.1
|
|
- torchvision=0.13.1
|
|
- numpy=1.23.1
|
|
- pip:
|
|
- albumentations==1.3.0
|
|
- opencv-python==4.6.0.66
|
|
- imageio==2.9.0
|
|
- imageio-ffmpeg==0.4.2
|
|
- pytorch-lightning==1.4.2
|
|
- omegaconf==2.1.1
|
|
- test-tube>=0.7.5
|
|
- streamlit==1.12.1
|
|
- einops==0.3.0
|
|
- transformers==4.19.2
|
|
- webdataset==0.2.5
|
|
- kornia==0.6
|
|
- open_clip_torch==2.0.2
|
|
- invisible-watermark>=0.1.5
|
|
- streamlit-drawable-canvas==0.8.0
|
|
- torchmetrics==0.6.0
|
|
- -e .
|