p1atdev commited on
Commit
a90cfcf
1 Parent(s): 34e6a75

doc: fix sample code

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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")