Unverified Commit 91925fbb authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

Fix callback type hints - no optional function argument (#2357)

replace type hints
parent 0db19da0
...@@ -257,7 +257,7 @@ class OnnxStableDiffusionInpaintPipeline(DiffusionPipeline): ...@@ -257,7 +257,7 @@ class OnnxStableDiffusionInpaintPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, np.ndarray], None]] = None, callback: Optional[Callable[[int, int, np.ndarray], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
): ):
r""" r"""
Function invoked when calling the pipeline for generation. Function invoked when calling the pipeline for generation.
......
...@@ -239,7 +239,7 @@ class OnnxStableDiffusionInpaintPipelineLegacy(DiffusionPipeline): ...@@ -239,7 +239,7 @@ class OnnxStableDiffusionInpaintPipelineLegacy(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, np.ndarray], None]] = None, callback: Optional[Callable[[int, int, np.ndarray], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
**kwargs, **kwargs,
): ):
r""" r"""
......
...@@ -472,7 +472,7 @@ class StableDiffusionPipeline(DiffusionPipeline): ...@@ -472,7 +472,7 @@ class StableDiffusionPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
cross_attention_kwargs: Optional[Dict[str, Any]] = None, cross_attention_kwargs: Optional[Dict[str, Any]] = None,
): ):
r""" r"""
......
...@@ -497,7 +497,7 @@ class StableDiffusionDepth2ImgPipeline(DiffusionPipeline): ...@@ -497,7 +497,7 @@ class StableDiffusionDepth2ImgPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
): ):
r""" r"""
Function invoked when calling the pipeline for generation. Function invoked when calling the pipeline for generation.
......
...@@ -274,7 +274,7 @@ class StableDiffusionImageVariationPipeline(DiffusionPipeline): ...@@ -274,7 +274,7 @@ class StableDiffusionImageVariationPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
): ):
r""" r"""
Function invoked when calling the pipeline for generation. Function invoked when calling the pipeline for generation.
......
...@@ -543,7 +543,7 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline): ...@@ -543,7 +543,7 @@ class StableDiffusionImg2ImgPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
**kwargs, **kwargs,
): ):
r""" r"""
......
...@@ -609,7 +609,7 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline): ...@@ -609,7 +609,7 @@ class StableDiffusionInpaintPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
): ):
r""" r"""
Function invoked when calling the pipeline for generation. Function invoked when calling the pipeline for generation.
......
...@@ -495,7 +495,7 @@ class StableDiffusionInpaintPipelineLegacy(DiffusionPipeline): ...@@ -495,7 +495,7 @@ class StableDiffusionInpaintPipelineLegacy(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
**kwargs, **kwargs,
): ):
r""" r"""
......
...@@ -141,7 +141,7 @@ class StableDiffusionInstructPix2PixPipeline(DiffusionPipeline): ...@@ -141,7 +141,7 @@ class StableDiffusionInstructPix2PixPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
): ):
r""" r"""
Function invoked when calling the pipeline for generation. Function invoked when calling the pipeline for generation.
......
...@@ -364,7 +364,7 @@ class StableDiffusionKDiffusionPipeline(DiffusionPipeline): ...@@ -364,7 +364,7 @@ class StableDiffusionKDiffusionPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
): ):
r""" r"""
Function invoked when calling the pipeline for generation. Function invoked when calling the pipeline for generation.
......
...@@ -291,7 +291,7 @@ class StableDiffusionLatentUpscalePipeline(DiffusionPipeline): ...@@ -291,7 +291,7 @@ class StableDiffusionLatentUpscalePipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
): ):
r""" r"""
Function invoked when calling the pipeline for generation. Function invoked when calling the pipeline for generation.
......
...@@ -388,7 +388,7 @@ class StableDiffusionUpscalePipeline(DiffusionPipeline): ...@@ -388,7 +388,7 @@ class StableDiffusionUpscalePipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
): ):
r""" r"""
Function invoked when calling the pipeline for generation. Function invoked when calling the pipeline for generation.
......
...@@ -620,7 +620,7 @@ class StableUnCLIPPipeline(DiffusionPipeline): ...@@ -620,7 +620,7 @@ class StableUnCLIPPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
cross_attention_kwargs: Optional[Dict[str, Any]] = None, cross_attention_kwargs: Optional[Dict[str, Any]] = None,
noise_level: int = 0, noise_level: int = 0,
# prior args # prior args
......
...@@ -587,7 +587,7 @@ class StableUnCLIPImg2ImgPipeline(DiffusionPipeline): ...@@ -587,7 +587,7 @@ class StableUnCLIPImg2ImgPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
cross_attention_kwargs: Optional[Dict[str, Any]] = None, cross_attention_kwargs: Optional[Dict[str, Any]] = None,
noise_level: int = 0, noise_level: int = 0,
image_embeds: Optional[torch.FloatTensor] = None, image_embeds: Optional[torch.FloatTensor] = None,
......
...@@ -512,7 +512,7 @@ class StableDiffusionPipelineSafe(DiffusionPipeline): ...@@ -512,7 +512,7 @@ class StableDiffusionPipelineSafe(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
sld_guidance_scale: Optional[float] = 1000, sld_guidance_scale: Optional[float] = 1000,
sld_warmup_steps: Optional[int] = 10, sld_warmup_steps: Optional[int] = 10,
sld_threshold: Optional[float] = 0.01, sld_threshold: Optional[float] = 0.01,
......
...@@ -95,7 +95,7 @@ class VersatileDiffusionPipeline(DiffusionPipeline): ...@@ -95,7 +95,7 @@ class VersatileDiffusionPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
): ):
r""" r"""
Function invoked when calling the pipeline for generation. Function invoked when calling the pipeline for generation.
...@@ -211,7 +211,7 @@ class VersatileDiffusionPipeline(DiffusionPipeline): ...@@ -211,7 +211,7 @@ class VersatileDiffusionPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
): ):
r""" r"""
Function invoked when calling the pipeline for generation. Function invoked when calling the pipeline for generation.
...@@ -324,7 +324,7 @@ class VersatileDiffusionPipeline(DiffusionPipeline): ...@@ -324,7 +324,7 @@ class VersatileDiffusionPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
): ):
r""" r"""
Function invoked when calling the pipeline for generation. Function invoked when calling the pipeline for generation.
......
...@@ -419,7 +419,7 @@ class VersatileDiffusionDualGuidedPipeline(DiffusionPipeline): ...@@ -419,7 +419,7 @@ class VersatileDiffusionDualGuidedPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
**kwargs, **kwargs,
): ):
r""" r"""
......
...@@ -271,7 +271,7 @@ class VersatileDiffusionImageVariationPipeline(DiffusionPipeline): ...@@ -271,7 +271,7 @@ class VersatileDiffusionImageVariationPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
**kwargs, **kwargs,
): ):
r""" r"""
......
...@@ -353,7 +353,7 @@ class VersatileDiffusionTextToImagePipeline(DiffusionPipeline): ...@@ -353,7 +353,7 @@ class VersatileDiffusionTextToImagePipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
**kwargs, **kwargs,
): ):
r""" r"""
......
...@@ -176,7 +176,7 @@ class VQDiffusionPipeline(DiffusionPipeline): ...@@ -176,7 +176,7 @@ class VQDiffusionPipeline(DiffusionPipeline):
output_type: Optional[str] = "pil", output_type: Optional[str] = "pil",
return_dict: bool = True, return_dict: bool = True,
callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None, callback: Optional[Callable[[int, int, torch.FloatTensor], None]] = None,
callback_steps: Optional[int] = 1, callback_steps: int = 1,
) -> Union[ImagePipelineOutput, Tuple]: ) -> Union[ImagePipelineOutput, Tuple]:
""" """
Function invoked when calling the pipeline for generation. Function invoked when calling the pipeline for generation.
......
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