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 ...@@ -30,7 +30,7 @@ cd stablediffussion/docker
docker build --no-cache -t stablediffussion:test . 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 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 cd stablediffussion
## 下载checkpoint model ## 下载checkpoint model
...@@ -52,7 +52,10 @@ git clone https://huggingface.co/runwayml/stable-diffusion-v1-5 ...@@ -52,7 +52,10 @@ git clone https://huggingface.co/runwayml/stable-diffusion-v1-5
git clone https://huggingface.co/CompVis/stable-diffusion-safety-checker git clone https://huggingface.co/CompVis/stable-diffusion-safety-checker
``` ```
## 推理: ## 数据集
## 推理
### 1、运行checkpoint version示例: ### 1、运行checkpoint version示例:
...@@ -129,10 +132,10 @@ image.save("astronaut_rides_horse.png") ...@@ -129,10 +132,10 @@ image.save("astronaut_rides_horse.png")
## 应用场景 ## 应用场景
### 算法类别 ### 算法类别
`文图生成` `以文生图`
### 热点应用行业 ### 热点应用行业
`绘画,动漫,3D人物模型` `绘画,动漫`
## 源码仓库及问题反馈 ## 源码仓库及问题反馈
http://developer.hpccube.com/codes/modelzoo/stablediffussion.git http://developer.hpccube.com/codes/modelzoo/stablediffussion.git
......
...@@ -5,6 +5,6 @@ modelName=stablediffusion_pytorch ...@@ -5,6 +5,6 @@ modelName=stablediffusion_pytorch
# 模型描述 # 模型描述
modelDescription=基于pytorch框架的stablediffusion文图生成推理示例 modelDescription=基于pytorch框架的stablediffusion文图生成推理示例
# 应用场景(多个标签以英文逗号分割) # 应用场景(多个标签以英文逗号分割)
appScenario=推理,文图生成,绘画,动漫,3D人物模型 appScenario=推理,以文生图,绘画,动漫
# 框架类型(多个标签以英文逗号分割) # 框架类型(多个标签以英文逗号分割)
frameType=pytorch 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