doc: fix sample code
Browse files
README.md
CHANGED
@@ -35,7 +35,7 @@ In version 1.2.1, the following special tags can be used
|
|
35 |
|
36 |
# 🧨 Diffusers
|
37 |
|
38 |
-
```
|
39 |
from diffusers import StableDiffusionPipeline
|
40 |
import torch
|
41 |
|
@@ -43,7 +43,7 @@ model_id = "p1atdev/plat-diffusion"
|
|
43 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
44 |
pipe = pipe.to("cuda")
|
45 |
|
46 |
-
prompt = "masterpiece, 1girl, "
|
47 |
image = pipe(prompt).images[0]
|
48 |
|
49 |
image.save("girl.png")
|
|
|
35 |
|
36 |
# 🧨 Diffusers
|
37 |
|
38 |
+
```py
|
39 |
from diffusers import StableDiffusionPipeline
|
40 |
import torch
|
41 |
|
|
|
43 |
pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
|
44 |
pipe = pipe.to("cuda")
|
45 |
|
46 |
+
prompt = "masterpiece, 1girl, surrounded by many flowers"
|
47 |
image = pipe(prompt).images[0]
|
48 |
|
49 |
image.save("girl.png")
|