Unverified Commit f9522d82 authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

[StableDiffusionPipeline] use default params in __call__ (#196)

use default params in __call__
parent 80e0c8ba
......@@ -60,7 +60,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
prompt = "a photo of an astronaut riding a horse on mars"
with autocast("cuda"):
image = pipe(prompt, width=512, guidance_scale=7.5)["sample"][0]
image = pipe(prompt)["sample"][0]
image.save("astronaut_rides_horse.png")
```
......
......@@ -31,7 +31,7 @@ class StableDiffusionPipeline(DiffusionPipeline):
height: Optional[int] = 512,
width: Optional[int] = 512,
num_inference_steps: Optional[int] = 50,
guidance_scale: Optional[float] = 1.0,
guidance_scale: Optional[float] = 7.5,
eta: Optional[float] = 0.0,
generator: Optional[torch.Generator] = None,
torch_device: Optional[Union[str, torch.device]] = None,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment