EdgeDiffuse_r4 / README.md
ChenHe727's picture
Update README.md
22dc910 verified

To Start

from diffusers import StableDiffusionPipeline
import torch

pipe = StableDiffusionPipeline.from_pretrained("ChenHe727/EdgeDiffuse_r4", torch_dtype=torch.float16)
pipe = pipe.to("cuda")
image = pipe("a photo of a cat", num_inference_steps=4, guidance_scale=0.0).images[0]