"tests/nn/git@developer.sourcefind.cn:OpenDAS/fairscale.git" did not exist on "7d7edf6d37576fb6eda65db6db43fda54a7f06ba"
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( ...@@ -60,7 +60,7 @@ pipe = StableDiffusionPipeline.from_pretrained(
prompt = "a photo of an astronaut riding a horse on mars" prompt = "a photo of an astronaut riding a horse on mars"
with autocast("cuda"): 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") image.save("astronaut_rides_horse.png")
``` ```
......
...@@ -31,7 +31,7 @@ class StableDiffusionPipeline(DiffusionPipeline): ...@@ -31,7 +31,7 @@ class StableDiffusionPipeline(DiffusionPipeline):
height: Optional[int] = 512, height: Optional[int] = 512,
width: Optional[int] = 512, width: Optional[int] = 512,
num_inference_steps: Optional[int] = 50, num_inference_steps: Optional[int] = 50,
guidance_scale: Optional[float] = 1.0, guidance_scale: Optional[float] = 7.5,
eta: Optional[float] = 0.0, eta: Optional[float] = 0.0,
generator: Optional[torch.Generator] = None, generator: Optional[torch.Generator] = None,
torch_device: Optional[Union[str, torch.device]] = 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