Update README.md
Browse files
README.md
CHANGED
|
@@ -71,9 +71,9 @@ import paddle
|
|
| 71 |
from paddle.amp import auto_cast as autocast
|
| 72 |
from diffusers_paddle import StableDiffusionPipeline
|
| 73 |
|
| 74 |
-
model_id = "
|
| 75 |
|
| 76 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id
|
| 77 |
|
| 78 |
prompt = "a photo of an astronaut riding a horse on mars"
|
| 79 |
with autocast(True):
|
|
@@ -89,7 +89,7 @@ If you are limited by GPU memory and have less than 10GB of GPU RAM available, p
|
|
| 89 |
```py
|
| 90 |
import paddle
|
| 91 |
|
| 92 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id, paddle_dtype=paddle.float16
|
| 93 |
|
| 94 |
prompt = "a photo of an astronaut riding a horse on mars"
|
| 95 |
with autocast(True):
|
|
@@ -106,7 +106,7 @@ from paddle.amp import auto_cast as autocast
|
|
| 106 |
model_id = "junnyu/stable-diffusion-v1-4-paddle"
|
| 107 |
# Use the K-LMS scheduler here instead
|
| 108 |
scheduler = LMSDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000)
|
| 109 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler
|
| 110 |
|
| 111 |
prompt = "a photo of an astronaut riding a horse on mars"
|
| 112 |
with autocast(True):
|
|
|
|
| 71 |
from paddle.amp import auto_cast as autocast
|
| 72 |
from diffusers_paddle import StableDiffusionPipeline
|
| 73 |
|
| 74 |
+
model_id = "junnyu/stable-diffusion-v1-4-paddle"
|
| 75 |
|
| 76 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id)
|
| 77 |
|
| 78 |
prompt = "a photo of an astronaut riding a horse on mars"
|
| 79 |
with autocast(True):
|
|
|
|
| 89 |
```py
|
| 90 |
import paddle
|
| 91 |
|
| 92 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, paddle_dtype=paddle.float16)
|
| 93 |
|
| 94 |
prompt = "a photo of an astronaut riding a horse on mars"
|
| 95 |
with autocast(True):
|
|
|
|
| 106 |
model_id = "junnyu/stable-diffusion-v1-4-paddle"
|
| 107 |
# Use the K-LMS scheduler here instead
|
| 108 |
scheduler = LMSDiscreteScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", num_train_timesteps=1000)
|
| 109 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, scheduler=scheduler)
|
| 110 |
|
| 111 |
prompt = "a photo of an astronaut riding a horse on mars"
|
| 112 |
with autocast(True):
|