Skip to content

Commit f9522d8

Browse files
authored
[StableDiffusionPipeline] use default params in __call__ (huggingface#196)
use default params in __call__
1 parent 80e0c8b commit f9522d8

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
6060

6161
prompt = "a photo of an astronaut riding a horse on mars"
6262
with autocast("cuda"):
63-
image = pipe(prompt, width=512, guidance_scale=7.5)["sample"][0]
63+
image = pipe(prompt)["sample"][0]
6464

6565
image.save("astronaut_rides_horse.png")
6666
```

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ def __call__(
3131
height: Optional[int] = 512,
3232
width: Optional[int] = 512,
3333
num_inference_steps: Optional[int] = 50,
34-
guidance_scale: Optional[float] = 1.0,
34+
guidance_scale: Optional[float] = 7.5,
3535
eta: Optional[float] = 0.0,
3636
generator: Optional[torch.Generator] = None,
3737
torch_device: Optional[Union[str, torch.device]] = None,

0 commit comments

Comments
 (0)