Fix type annotations on StableDiffusionPipeline.__call__ (#682)

Fixed type annotations on StableDiffusionPipeline::__call__

Fix type annotations on StableDiffusionPipeline.call (#682)
Fixed type annotations on StableDiffusionPipeline::__call__
7d0ba592 · Krishna Penukonda · GitHub · 249b36cc · 7d0ba592
Unverified Commit 7d0ba592 authored Oct 03, 2022 by Krishna Penukonda Committed by GitHub Oct 03, 2022
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 5 deletions

src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py ...s/pipelines/stable_diffusion/pipeline_stable_diffusion.py +5 -5

No files found.
--- a/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
+++ b/src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
@@ -113,11 +113,11 @@ class StableDiffusionPipeline(DiffusionPipeline):
    def __call__(
        self,
        prompt: Union[str, List[str]],
-        height: Optional[int] = 512,
-        width: Optional[int] = 512,
-        num_inference_steps: Optional[int] = 50,
-        guidance_scale: Optional[float] = 7.5,
-        eta: Optional[float] = 0.0,
+        height: int = 512,
+        width: int = 512,
+        num_inference_steps: int = 50,
+        guidance_scale: float = 7.5,
+        eta: float = 0.0,
        generator: Optional[torch.Generator] = None,
        latents: Optional[torch.FloatTensor] = None,
        output_type: Optional[str] = "pil",