SkyWork
commited on
Commit
·
07352f8
1
Parent(s):
b049bdb
Update README.md
Browse files
README.md
CHANGED
@@ -130,13 +130,11 @@ with torch.no_grad():
|
|
130 |
|
131 |
## 测试用例
|
132 |
|
133 |
-
模型下载地址 [SkyPaint-v1.0](https://sai-hk.oss-cn-hongkong.aliyuncs.com/zb/skypaint-v-1.0.zip?OSSAccessKeyId=LTAI5tHuxqp63n5qw5eeB6Ji&Expires=1673528832&Signature=4PTeknRoXuHWmeQHXqgu8kB0q%2Bw%3D)
|
134 |
-
|
135 |
```py
|
136 |
from diffusers import StableDiffusionPipeline
|
137 |
|
138 |
device = 'cuda'
|
139 |
-
pipe = StableDiffusionPipeline.from_pretrained("
|
140 |
|
141 |
prompts = [
|
142 |
'机械狗',
|
|
|
130 |
|
131 |
## 测试用例
|
132 |
|
|
|
|
|
133 |
```py
|
134 |
from diffusers import StableDiffusionPipeline
|
135 |
|
136 |
device = 'cuda'
|
137 |
+
pipe = StableDiffusionPipeline.from_pretrained("SkyWork/SkyPaint").to(device)
|
138 |
|
139 |
prompts = [
|
140 |
'机械狗',
|