Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-08-26 15:16:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

RUF013 updates

Browse Source
This commit is contained in:
awsr
2026-03-24 06:15:26 -07:00
parent 1003926646
commit 876e3b9897
3 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/apg/pipeline_stable_cascade_prior_apg.py
+2 -2
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.
modules/apg/pipeline_stable_diffision_xl_apg.py
+4 -4
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"""
modules/apg/pipeline_stable_diffusion_apg.py
+4 -4
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"""