ChenHe727 commited on
Commit
9360ee3
·
verified ·
1 Parent(s): 9ea5cb3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -0
README.md CHANGED
@@ -1,9 +1,11 @@
1
  ### To Start
2
 
3
  from diffusers import StableDiffusionPipeline
 
4
  import torch
5
 
6
  pipe = StableDiffusionPipeline.from_pretrained("ChenHe727/EdgeDiffuse_r4", torch_dtype=torch.float16)
 
7
  pipe = pipe.to("cuda")
8
 
9
  image = pipe("a photo of a cat", num_inference_steps=4, guidance_scale=0.0).images[0]
 
1
  ### To Start
2
 
3
  from diffusers import StableDiffusionPipeline
4
+
5
  import torch
6
 
7
  pipe = StableDiffusionPipeline.from_pretrained("ChenHe727/EdgeDiffuse_r4", torch_dtype=torch.float16)
8
+
9
  pipe = pipe.to("cuda")
10
 
11
  image = pipe("a photo of a cat", num_inference_steps=4, guidance_scale=0.0).images[0]