Commit 82eab2a1 authored by lijian6's avatar lijian6
Browse files

Modify README.md

parent 7f762fd0
...@@ -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数据库子集的512x512图像上训练潜在扩散模型。与谷歌的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
``` ```
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment