Skip to content

GitLab

"...text-generation-inference.git" did not exist on "b4024edd4549ab647b02b8619b4072e33e64f1f9"
Unverified Commit 7d0ba592 authored by Krishna Penukonda's avatar Krishna Penukonda Committed by GitHub
Browse files

Fix type annotations on StableDiffusionPipeline.__call__ (#682) 

Fixed type annotations on StableDiffusionPipeline::__call__
parent 249b36cc
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 5 deletions
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
View file @ 7d0ba592
...@@ -113,11 +113,11 @@ class StableDiffusionPipeline(DiffusionPipeline): ...@@ -113,11 +113,11 @@ class StableDiffusionPipeline(DiffusionPipeline):
def __call__( def __call__(
self, self,
prompt: Union[str, List[str]], prompt: Union[str, List[str]],
height: Optional[int] = 512, height: int = 512,
width: Optional[int] = 512, width: int = 512,
num_inference_steps: Optional[int] = 50, num_inference_steps: int = 50,
guidance_scale: Optional[float] = 7.5, guidance_scale: float = 7.5,
eta: Optional[float] = 0.0, eta: float = 0.0,
generator: Optional[torch.Generator] = None, generator: Optional[torch.Generator] = None,
latents: Optional[torch.FloatTensor] = None, latents: Optional[torch.FloatTensor] = None,
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
......
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