1580158
/
automatic
mirror of https://github.com/vladmandic/automatic
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

RUF013 updates

pull/4706/head
awsr 2026-03-24 06:15:26 -07:00
parent 1003926646
commit 876e3b9897
No known key found for this signature in database
3 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
modules/apg/pipeline_stable_cascade_prior_apg.py
View File

@ -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.

8
modules/apg/pipeline_stable_diffision_xl_apg.py
View File

@ -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"""

8
modules/apg/pipeline_stable_diffusion_apg.py
View File

@ -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"""