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