Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
ModelZoo
stable-diffusion_pytorch
Commits
82eab2a1
Commit
82eab2a1
authored
Aug 22, 2023
by
lijian6
Browse files
Modify README.md
parent
7f762fd0
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
15 additions
and
3 deletions
+15
-3
README.md
README.md
+15
-3
No files found.
README.md
View file @
82eab2a1
...
@@ -2,7 +2,7 @@
...
@@ -2,7 +2,7 @@
## 模型介绍
## 模型介绍
Stable Diffusion是一种
潜在的文本到图像
扩散型。我们能够
在
LAION-512B数据库子集的512x5图像上训练潜在扩散模型。与谷歌的Imagen类似,该模型使用
冻结的
CLIP ViT-L/14
文本编码器根据文本提示对模型进行调节。凭借其860M UNet和123M文本编码器,该模型相对轻巧,可以在
至
少具有10GB VRAM的GPU上运行。请参阅下面的此部分和模型卡。
Stable Diffusion是一种
文图生成潜在
扩散型。我们能够
使用
LAION-512B数据库子集的512x5
12
图像上训练潜在扩散模型。与谷歌的Imagen类似,该模型使用
frozen
CLIP ViT-L/14文本编码器根据文本提示对模型进行调节。凭借其860M UNet和123M文本编码器,该模型相对轻巧,可以在少
至
具有10GB VRAM的GPU上运行。请参阅下面的此部分和模型卡。
## 环境配置
## 环境配置
...
@@ -10,7 +10,7 @@ Stable Diffusion是一种潜在的文本到图像扩散型。我们能够在LAIO
...
@@ -10,7 +10,7 @@ Stable Diffusion是一种潜在的文本到图像扩散型。我们能够在LAIO
在光源可拉取推理的docker镜像,拉取方式如下:
在光源可拉取推理的docker镜像,拉取方式如下:
```
```
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:stable-diffusion
docker pull image.sourcefind.cn:5000/dcu/admin/base/custom:stable-diffusion
_pytorch1.10.0-centos7.6-dtk-22.10.1-py38
```
```
2、使用conda环境:
2、使用conda环境:
...
@@ -24,14 +24,26 @@ conda activate ldm
...
@@ -24,14 +24,26 @@ conda activate ldm
## 下载Stable Diffusion模型
## 下载Stable Diffusion模型
```
```
cd stablediffussion
### checkpoint version
### checkpoint version
git lfs install
git lfs install
git clone https://huggingface.co/CompVis/stable-diffusion-v-1-1-original
git clone https://huggingface.co/CompVis/stable-diffusion-v-1-1-original
git clone https://huggingface.co/CompVis/stable-diffusion-v-1-2-original
git clone https://huggingface.co/CompVis/stable-diffusion-v-1-3-original
git clone https://huggingface.co/CompVis/stable-diffusion-v-1-4-original
### diffusers version
### diffusers version
git clone https://huggingface.co/CompVis/stable-diffusion-v1-1
git clone https://huggingface.co/CompVis/stable-diffusion-v1-1
git clone https://huggingface.co/CompVis/stable-diffusion-v1-2
git clone https://huggingface.co/CompVis/stable-diffusion-v1-3
git clone https://huggingface.co/CompVis/stable-diffusion-v1-4
git clone https://huggingface.co/runwayml/stable-diffusion-v1-5
### safety-checker
git clone https://huggingface.co/CompVis/stable-diffusion-safety-checker
```
```
## 运行示例:
## 运行示例:
### 运行checkpoint version示例:
### 运行checkpoint version示例:
...
@@ -44,7 +56,7 @@ ln -s <path/to/model.ckpt> models/ldm/stable-diffusion-v1/model.ckpt
...
@@ -44,7 +56,7 @@ ln -s <path/to/model.ckpt> models/ldm/stable-diffusion-v1/model.ckpt
运行:
运行:
```
```
python scripts/txt2img.py --prompt "a photograph of an astronaut riding a horse" --plms
python scripts/txt2img.py --prompt "a photograph of an astronaut riding a horse" --plms
```
```
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment