kanshan/diffusers
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix type annotations on StableDiffusionPipeline.__call__ (#682)

Browse Source 
Fixed type annotations on StableDiffusionPipeline::__call__
This commit is contained in:
Krishna Penukonda
2022-10-03 21:38:24 +08:00
committed by GitHub
parent 249b36cc38
commit 7d0ba5921b
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/diffusers/pipelines/stable_diffusion/pipeline_stable_diffusion.py
+5 -5
View File
@@ -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",