mirror of
https://github.com/Stability-AI/stablediffusion.git
synced 2024-12-22 23:55:00 +00:00
31 lines
816 B
YAML
31 lines
816 B
YAML
|
build:
|
||
|
|
gpu: true
|
||
|
|
cuda: "11.3"
|
||
|
|
python_version: 3.8
|
||
|
|
system_packages:
|
||
|
|
- libgl1-mesa-glx
|
||
|
|
- libglib2.0-0
|
||
|
|
python_packages:
|
||
|
|
- torch==1.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
|
||
|
|
# - torchvision==0.13.1+cu113--extra-index-url https://download.pytorch.org/whl/cu113
|
||
|
|
- albumentations==1.3.0
|
||
|
|
- open-clip-torch==2.7.0
|
||
|
|
- opencv-python==4.6.0.66
|
||
|
|
- imageio==2.22.2
|
||
|
|
- opencv-python==4.6.0.66
|
||
|
|
- imageio-ffmpeg==0.4.2
|
||
|
|
- pytorch-lightning==1.7.7
|
||
|
|
- omegaconf==2.1.1
|
||
|
|
- test-tube==0.7.5
|
||
|
|
- 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.10.2
|
||
|
|
|
||
|
|
|
||
|
|
predict: "scripts/replicate/predict.py:Predictor"
|