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):
|