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

patrickvonplaten · web-flow · commit 6c357298d6ea · 2023-02-14T14:35:05.000-08:00
replace type hints
diff --git a/pipelines/alt_diffusion/pipeline_alt_diffusion.py b/pipelines/alt_diffusion/pipeline_alt_diffusion.py
@@ -476,7 +476,7 @@ def __call__(
         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"""
diff --git a/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py b/pipelines/alt_diffusion/pipeline_alt_diffusion_img2img.py
@@ -534,7 +534,7 @@ def __call__(
         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"""
diff --git a/pipelines/paint_by_example/pipeline_paint_by_example.py b/pipelines/paint_by_example/pipeline_paint_by_example.py
@@ -400,7 +400,7 @@ def __call__(
         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.
diff --git a/pipelines/stable_diffusion/pipeline_cycle_diffusion.py b/pipelines/stable_diffusion/pipeline_cycle_diffusion.py
@@ -554,7 +554,7 @@ def __call__(
         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"""
diff --git a/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py b/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion.py
@@ -203,7 +203,7 @@ def __call__(
         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,
     ):
         if isinstance(prompt, str):
             batch_size = 1
diff --git a/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_img2img.py b/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_img2img.py
@@ -252,7 +252,7 @@ def __call__(
         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"""
diff --git a/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint.py b/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint.py
@@ -257,7 +257,7 @@ def __call__(
         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.
diff --git a/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint_legacy.py b/pipelines/stable_diffusion/pipeline_onnx_stable_diffusion_inpaint_legacy.py
@@ -239,7 +239,7 @@ def __call__(
         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"""
diff --git a/pipelines/stable_diffusion/pipeline_stable_diffusion.py b/pipelines/stable_diffusion/pipeline_stable_diffusion.py
@@ -472,7 +472,7 @@ def __call__(
         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"""
diff --git a/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py b/pipelines/stable_diffusion/pipeline_stable_diffusion_depth2img.py
@@ -497,7 +497,7 @@ def __call__(
         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.
diff --git a/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py b/pipelines/stable_diffusion/pipeline_stable_diffusion_image_variation.py
@@ -274,7 +274,7 @@ def __call__(
         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.
diff --git a/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py b/pipelines/stable_diffusion/pipeline_stable_diffusion_img2img.py
@@ -543,7 +543,7 @@ def __call__(
         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"""
diff --git a/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py b/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint.py
@@ -609,7 +609,7 @@ def __call__(
         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.
diff --git a/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py b/pipelines/stable_diffusion/pipeline_stable_diffusion_inpaint_legacy.py
@@ -495,7 +495,7 @@ def __call__(
         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"""
diff --git a/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py b/pipelines/stable_diffusion/pipeline_stable_diffusion_instruct_pix2pix.py
@@ -141,7 +141,7 @@ def __call__(
         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.
diff --git a/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py b/pipelines/stable_diffusion/pipeline_stable_diffusion_k_diffusion.py
@@ -364,7 +364,7 @@ def __call__(
         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.
diff --git a/pipelines/stable_diffusion/pipeline_stable_diffusion_latent_upscale.py b/pipelines/stable_diffusion/pipeline_stable_diffusion_latent_upscale.py
@@ -291,7 +291,7 @@ def __call__(
         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.
diff --git a/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py b/pipelines/stable_diffusion/pipeline_stable_diffusion_upscale.py
@@ -388,7 +388,7 @@ def __call__(
         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.
diff --git a/pipelines/stable_diffusion/pipeline_stable_unclip.py b/pipelines/stable_diffusion/pipeline_stable_unclip.py
@@ -620,7 +620,7 @@ def __call__(
         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
diff --git a/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py b/pipelines/stable_diffusion/pipeline_stable_unclip_img2img.py
@@ -587,7 +587,7 @@ def __call__(
         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,
diff --git a/pipelines/stable_diffusion_safe/pipeline_stable_diffusion_safe.py b/pipelines/stable_diffusion_safe/pipeline_stable_diffusion_safe.py
@@ -512,7 +512,7 @@ def __call__(
         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,
diff --git a/pipelines/versatile_diffusion/pipeline_versatile_diffusion.py b/pipelines/versatile_diffusion/pipeline_versatile_diffusion.py
@@ -95,7 +95,7 @@ def image_variation(
         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 @@ def text_to_image(
         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 @@ def dual_guided(
         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.
diff --git a/pipelines/versatile_diffusion/pipeline_versatile_diffusion_dual_guided.py b/pipelines/versatile_diffusion/pipeline_versatile_diffusion_dual_guided.py
@@ -419,7 +419,7 @@ def __call__(
         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"""
diff --git a/pipelines/versatile_diffusion/pipeline_versatile_diffusion_image_variation.py b/pipelines/versatile_diffusion/pipeline_versatile_diffusion_image_variation.py
@@ -271,7 +271,7 @@ def __call__(
         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"""
diff --git a/pipelines/versatile_diffusion/pipeline_versatile_diffusion_text_to_image.py b/pipelines/versatile_diffusion/pipeline_versatile_diffusion_text_to_image.py
@@ -353,7 +353,7 @@ def __call__(
         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"""
diff --git a/pipelines/vq_diffusion/pipeline_vq_diffusion.py b/pipelines/vq_diffusion/pipeline_vq_diffusion.py
@@ -176,7 +176,7 @@ def __call__(
         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.
