Commit d7333246 authored by lijian6's avatar lijian6
Browse files

Update README.md


Signed-off-by: lijian6's avatarlijian <lijian6@sugon.com>
parent df7a7242
......@@ -30,7 +30,7 @@ cd stablediffussion/docker
docker build --no-cache -t stablediffussion:test .
docker run --rm --shm-size 10g --network=host --name=stablediffussion --privileged --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -v path_to_stablediffussion:/home/sd -it stablediffussion:test bash
```
## 下载Stable Diffusion模型
### 下载Stable Diffusion模型
```
cd stablediffussion
## 下载checkpoint model
......@@ -52,7 +52,10 @@ git clone https://huggingface.co/runwayml/stable-diffusion-v1-5
git clone https://huggingface.co/CompVis/stable-diffusion-safety-checker
```
## 推理:
## 数据集
## 推理
### 1、运行checkpoint version示例:
......@@ -129,10 +132,10 @@ image.save("astronaut_rides_horse.png")
## 应用场景
### 算法类别
`文图生成`
`以文生图`
### 热点应用行业
`绘画,动漫,3D人物模型`
`绘画,动漫`
## 源码仓库及问题反馈
http://developer.hpccube.com/codes/modelzoo/stablediffussion.git
......
......@@ -5,6 +5,6 @@ modelName=stablediffusion_pytorch
# 模型描述
modelDescription=基于pytorch框架的stablediffusion文图生成推理示例
# 应用场景(多个标签以英文逗号分割)
appScenario=推理,文图生成,绘画,动漫,3D人物模型
appScenario=推理,以文生图,绘画,动漫
# 框架类型(多个标签以英文逗号分割)
frameType=pytorch
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