diff --git a/modules/apg/pipeline_stable_cascade_prior_apg.py b/modules/apg/pipeline_stable_cascade_prior_apg.py index 6ffaf8089..c4747612d 100644 --- a/modules/apg/pipeline_stable_cascade_prior_apg.py +++ b/modules/apg/pipeline_stable_cascade_prior_apg.py @@ -372,7 +372,7 @@ class StableCascadePriorPipelineAPG(DiffusionPipeline): height: int = 1024, width: int = 1024, num_inference_steps: int = 20, - timesteps: list[float] = None, + timesteps: list[float] | None = None, guidance_scale: float = 4.0, negative_prompt: str | list[str] | None = None, prompt_embeds: torch.Tensor | None = None, @@ -386,7 +386,7 @@ class StableCascadePriorPipelineAPG(DiffusionPipeline): output_type: str | None = "pt", return_dict: bool = True, callback_on_step_end: Callable[[int, int, dict], None] | None = None, - callback_on_step_end_tensor_inputs: list[str] = None, + callback_on_step_end_tensor_inputs: list[str] | None = None, ): """ Function invoked when calling the pipeline for generation. diff --git a/modules/apg/pipeline_stable_diffision_xl_apg.py b/modules/apg/pipeline_stable_diffision_xl_apg.py index b09ae242f..5da74c299 100644 --- a/modules/apg/pipeline_stable_diffision_xl_apg.py +++ b/modules/apg/pipeline_stable_diffision_xl_apg.py @@ -793,13 +793,13 @@ class StableDiffusionXLPipelineAPG( @replace_example_docstring(EXAMPLE_DOC_STRING) def __call__( self, - prompt: str | list[str] = None, + prompt: str | list[str] | None = None, prompt_2: str | list[str] | None = None, height: int | None = None, width: int | None = None, num_inference_steps: int = 50, - timesteps: list[int] = None, - sigmas: list[float] = None, + timesteps: list[int] | None = None, + sigmas: list[float] | None = None, denoising_end: float | None = None, guidance_scale: float = 5.0, negative_prompt: str | list[str] | None = None, @@ -826,7 +826,7 @@ class StableDiffusionXLPipelineAPG( negative_target_size: tuple[int, int] | None = None, clip_skip: int | None = None, callback_on_step_end: Callable[[int, int, dict], None] | PipelineCallback | MultiPipelineCallbacks | None = None, - callback_on_step_end_tensor_inputs: list[str] = None, + callback_on_step_end_tensor_inputs: list[str] | None = None, **kwargs, ): r""" diff --git a/modules/apg/pipeline_stable_diffusion_apg.py b/modules/apg/pipeline_stable_diffusion_apg.py index ae1eb26e0..57aebafca 100644 --- a/modules/apg/pipeline_stable_diffusion_apg.py +++ b/modules/apg/pipeline_stable_diffusion_apg.py @@ -749,12 +749,12 @@ class StableDiffusionPipelineAPG( @replace_example_docstring(EXAMPLE_DOC_STRING) def __call__( self, - prompt: str | list[str] = None, + prompt: str | list[str] | None = None, height: int | None = None, width: int | None = None, num_inference_steps: int = 50, - timesteps: list[int] = None, - sigmas: list[float] = None, + timesteps: list[int] | None = None, + sigmas: list[float] | None = None, guidance_scale: float = 7.5, negative_prompt: str | list[str] | None = None, num_images_per_prompt: int | None = 1, @@ -771,7 +771,7 @@ class StableDiffusionPipelineAPG( guidance_rescale: float = 0.0, clip_skip: int | None = None, callback_on_step_end: Callable[[int, int, dict], None] | PipelineCallback | MultiPipelineCallbacks | None = None, - callback_on_step_end_tensor_inputs: list[str] = None, + callback_on_step_end_tensor_inputs: list[str] | None = None, **kwargs, ): r"""