Commit 7f762fd0 authored by lijian6's avatar lijian6
Browse files

Modify README.md

parent 1e2eee38
Pipeline #526 failed with stages
in 0 seconds
...@@ -23,13 +23,14 @@ conda activate ldm ...@@ -23,13 +23,14 @@ conda activate ldm
``` ```
## 下载Stable Diffusion模型 ## 下载Stable Diffusion模型
```
### 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
### diffusers version ### diffusers version
git clone https://huggingface.co/CompVis/stable-diffusion-v1-1 git clone https://huggingface.co/CompVis/stable-diffusion-v1-1
```
## 运行示例: ## 运行示例:
...@@ -84,14 +85,15 @@ optional arguments: ...@@ -84,14 +85,15 @@ optional arguments:
### 运行Diffusers示例: ### 运行Diffusers示例:
Diffusers.py: 下载diffusers, 使用Diffusers.py运行:
```py ```py
# make sure you're logged in with `huggingface-cli login` # make sure you're logged in with `huggingface-cli login`
import torch
from torch import autocast from torch import autocast
from diffusers import StableDiffusionPipeline from diffusers import StableDiffusionPipeline
pipe = StableDiffusionPipeline.from_pretrained( pipe = StableDiffusionPipeline.from_pretrained(
"CompVis/stable-diffusion-v1-4", "stable-diffusion-v1-4",
use_auth_token=True use_auth_token=True
).to("cuda") ).to("cuda")
......
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