Skip to content

Commit b2b455f

Browse files
committed
Add environment-mps
1 parent 82bdf6d commit b2b455f

File tree

3 files changed

+41
-0
lines changed

3 files changed

+41
-0
lines changed

.DS_Store

10 KB
Binary file not shown.

README.md

+7
Original file line numberDiff line numberDiff line change
@@ -48,9 +48,16 @@ Note that the way we connect layers is computational efficient. The original SD
4848

4949
First create a new conda environment
5050

51+
CUDA, CPU
52+
5153
conda env create -f environment.yaml
5254
conda activate control
5355

56+
MPS
57+
58+
conda env create -f environment-mps.yaml
59+
conda activate control
60+
5461
All models and detectors can be downloaded from [our Hugging Face page](https://huggingface.co/lllyasviel/ControlNet). Make sure that SD models are put in "ControlNet/models" and detectors are put in "ControlNet/annotator/ckpts". Make sure that you download all necessary pretrained weights and detector models from that Hugging Face page, including HED edge detection model, Midas depth estimation model, Openpose, and so on.
5562

5663
We provide 9 Gradio apps with these models.

environment-mps.yaml

+34
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,34 @@
1+
name: control
2+
channels:
3+
- pytorch
4+
- defaults
5+
dependencies:
6+
- python=3.8
7+
- pip
8+
- pytorch=1.12.1
9+
- torchvision=0.13.1
10+
- numpy=1.23.1
11+
- pip:
12+
- gradio==3.16.2
13+
- albumentations==1.3.0
14+
- opencv-contrib-python
15+
- imageio==2.9.0
16+
- imageio-ffmpeg==0.4.2
17+
- pytorch-lightning==1.5.0
18+
- omegaconf==2.1.1
19+
- test-tube>=0.7.5
20+
- streamlit==1.12.1
21+
- einops==0.3.0
22+
- transformers==4.19.2
23+
- webdataset==0.2.5
24+
- kornia==0.6
25+
- open_clip_torch==2.0.2
26+
- invisible-watermark>=0.1.5
27+
- streamlit-drawable-canvas==0.8.0
28+
- torchmetrics==0.6.0
29+
- timm==0.6.12
30+
- addict==2.4.0
31+
- yapf==0.32.0
32+
- prettytable==3.6.0
33+
- safetensors==0.2.7
34+
- basicsr==1.4.2

0 commit comments

Comments
 (0)