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 |
'机械狗',
|