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

PEP 484 scripts directory

pull/4711/head
awsr 2026-03-25 16:50:41 -07:00
parent a255e5bdcf
commit c3cce0ad76
No known key found for this signature in database
10 changed files with 34 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
scripts/ctrlx/__init__.py
View File

@ -136,13 +136,13 @@ class CtrlXStableDiffusionXLPipeline(StableDiffusionXLPipeline): # diffusers==0
@torch.no_grad()
def __call__(
self,
prompt: Union[str, List[str]] = None,
prompt: Union[str, List[str]] | None = None,
structure_prompt: Optional[Union[str, List[str]]] = None,
appearance_prompt: Optional[Union[str, List[str]]] = None,
structure_image: Optional[PipelineImageInput] = None,
appearance_image: Optional[PipelineImageInput] = None,
num_inference_steps: int = 50,
timesteps: List[int] = None,
timesteps: List[int] | None = None,
negative_prompt: Optional[Union[str, List[str]]] = None,
positive_prompt: Optional[Union[str, List[str]]] = None,
height: Optional[int] = None,
@ -172,9 +172,9 @@ class CtrlXStableDiffusionXLPipeline(StableDiffusionXLPipeline): # diffusers==0
return_dict: bool = True,
cross_attention_kwargs: Optional[Dict[str, Any]] = None,
guidance_rescale: float = 0.0,
original_size: Tuple[int, int] = None,
original_size: Tuple[int, int] | None = None,
crops_coords_top_left: Tuple[int, int] = (0, 0),
target_size: Tuple[int, int] = None,
target_size: Tuple[int, int] | None = None,
clip_skip: Optional[int] = None,
callback_on_step_end: Optional[Callable[[int, int, Dict], None]] = None,
callback_on_step_end_tensor_inputs: List[str] = ["latents"],

2
scripts/demofusion.py
View File

@ -498,7 +498,7 @@ class DemoFusionSDXLPipeline(DiffusionPipeline, FromSingleFileMixin, LoraLoaderM
@torch.no_grad()
def __call__(
self,
prompt: Union[str, List[str]] = None,
prompt: Union[str, List[str]] | None = None,
prompt_2: Optional[Union[str, List[str]]] = None,
height: Optional[int] = None,
width: Optional[int] = None,

8
scripts/differential_diffusion.py
View File

@ -631,7 +631,7 @@ class StableDiffusionXLDiffImg2ImgPipeline(DiffusionPipeline, FromSingleFileMixi
@replace_example_docstring(EXAMPLE_DOC_STRING)
def __call__(
self,
prompt: Union[str, List[str]] = None,
prompt: Union[str, List[str]] | None = None,
prompt_2: Optional[Union[str, List[str]]] = None,
image: Union[
torch.FloatTensor,
@ -662,9 +662,9 @@ class StableDiffusionXLDiffImg2ImgPipeline(DiffusionPipeline, FromSingleFileMixi
callback_steps: int = 1,
cross_attention_kwargs: Optional[Dict[str, Any]] = None,
guidance_rescale: float = 0.0,
original_size: Tuple[int, int] = None,
original_size: Tuple[int, int] | None = None,
crops_coords_top_left: Tuple[int, int] = (0, 0),
target_size: Tuple[int, int] = None,
target_size: Tuple[int, int] | None = None,
aesthetic_score: float = 6.0,
negative_aesthetic_score: float = 2.5,
map: torch.FloatTensor = None, # pylint: disable=redefined-builtin
@ -1643,7 +1643,7 @@ class StableDiffusionDiffImg2ImgPipeline(DiffusionPipeline):
@replace_example_docstring(EXAMPLE_DOC_STRING)
def __call__(
self,
prompt: Union[str, List[str]] = None,
prompt: Union[str, List[str]] | None = None,
image: Union[torch.FloatTensor, PIL.Image.Image] = None,
strength: float = 1,
num_inference_steps: Optional[int] = 50,

4
scripts/infiniteyou/pipeline_flux_infusenet.py
View File

@ -113,12 +113,12 @@ class FluxInfuseNetPipeline(FluxControlNetPipeline):
@torch.no_grad()
def __call__(
self,
prompt: Union[str, List[str]] = None,
prompt: Union[str, List[str]] | None = None,
prompt_2: Optional[Union[str, List[str]]] = None,
height: Optional[int] = None,
width: Optional[int] = None,
num_inference_steps: int = 28,
timesteps: List[int] = None,
timesteps: List[int] | None = None,
guidance_scale: float = 3.5,
id_image: PipelineImageInput = None,
controlnet_guidance_scale: float = 1.0,

2
scripts/infiniteyou_ext.py
View File

@ -60,7 +60,7 @@ class Script(scripts_manager.Script):
return [model, id_image, control_image, scale, start, end, id_guidance, control_guidance, restore]
def run(self, p: processing.StableDiffusionProcessing,
model: str = None,
model: str | None = None,
id_image: Image.Image = None,
control_image: Image.Image = None,
scale: float = 1.0,

2
scripts/mod/__init__.py
View File

@ -779,7 +779,7 @@ class StableDiffusionXLTilingPipeline(
@replace_example_docstring(EXAMPLE_DOC_STRING)
def __call__(
self,
prompt: Union[str, List[str]] = None,
prompt: Union[str, List[str]] | None = None,
height: Optional[int] = None,
width: Optional[int] = None,
num_inference_steps: int = 50,

2
scripts/nudenet/imageguard.py
View File

@ -93,7 +93,7 @@ model = None
processor = None
def image_guard(image, policy:str=None) -> str:
def image_guard(image, policy:str | None=None) -> str:
global model, processor # pylint: disable=global-statement
import json
from installer import install

6
scripts/pixelsmith/pixelsmith_pipeline.py
View File

@ -1088,12 +1088,12 @@ class PixelSmithXLPipeline(
@replace_example_docstring(EXAMPLE_DOC_STRING)
def __call__(
self,
prompt: Union[str, List[str]] = None,
prompt: Union[str, List[str]] | None = None,
prompt_2: Optional[Union[str, List[str]]] = None,
height: Optional[int] = None,
width: Optional[int] = None,
num_inference_steps: int = 50,
timesteps: List[int] = None,
timesteps: List[int] | None = None,
denoising_end: Optional[float] = None,
guidance_scale: float = 5.0,
#+#
@ -1101,7 +1101,7 @@ class PixelSmithXLPipeline(
pag_adaptive_scaling: float = 0.0,
pag_drop_rate: float = 0.5,
pag_applied_layers: List[str] = ['mid'], #['down', 'mid', 'up']
pag_applied_layers_index: List[str] = None, #['d4', 'd5', 'm0']
pag_applied_layers_index: List[str] | None = None, #['d4', 'd5', 'm0']
#+#
negative_prompt: Optional[Union[str, List[str]]] = None,
negative_prompt_2: Optional[Union[str, List[str]]] = None,

26
scripts/prompt_enhance.py
View File

@ -321,7 +321,7 @@ class Script(scripts_manager.Script):
from modules.sd_models_compile import compile_torch
self.llm = compile_torch(self.llm, apply_to_components=False, op="LLM")
def load(self, name:str=None, model_repo:str=None, model_gguf:str=None, model_type:str=None, model_file:str=None):
def load(self, name:str | None=None, model_repo:str | None=None, model_gguf:str | None=None, model_type:str | None=None, model_file:str | None=None):
# Strip symbols from display name if present
name = get_model_repo_from_display(name) if name else self.options.default
if self.busy:
@ -535,21 +535,21 @@ class Script(scripts_manager.Script):
return current_image
def enhance(self,
model: str=None,
prompt:str=None,
system:str=None,
prefix:str=None,
suffix:str=None,
sample:bool=None,
tokens:int=None,
temperature:float=None,
penalty:float=None,
top_k:int=None,
top_p:float=None,
model: str | None=None,
prompt:str | None=None,
system:str | None=None,
prefix:str | None=None,
suffix:str | None=None,
sample:bool | None=None,
tokens:int | None=None,
temperature:float | None=None,
penalty:float | None=None,
top_k:int | None=None,
top_p:float | None=None,
thinking:bool=False,
seed:int=-1,
image=None,
nsfw:bool=None,
nsfw:bool | None=None,
use_vision:bool=True,
prefill:str='',
keep_prefill:bool=False,

8
scripts/softfill.py
View File

@ -928,14 +928,14 @@ class StableDiffusionXLSoftFillPipeline(
@replace_example_docstring(EXAMPLE_DOC_STRING)
def __call__(
self,
prompt: Union[str, List[str]] = None,
prompt: Union[str, List[str]] | None = None,
prompt_2: Optional[Union[str, List[str]]] = None,
image: Image.Image = None,
mask: Image.Image = None,
noise_fill_image: bool = True, # Adds noise to the image at the masks >0.8 area.
strength: float = 0.3,
num_inference_steps: int = 50,
timesteps: List[int] = None,
timesteps: List[int] | None = None,
denoising_start: Optional[float] = None,
denoising_end: Optional[float] = None,
guidance_scale: float = 5.0,
@ -955,9 +955,9 @@ class StableDiffusionXLSoftFillPipeline(
return_dict: bool = True,
cross_attention_kwargs: Optional[Dict[str, Any]] = None,
guidance_rescale: float = 0.0,
original_size: Tuple[int, int] = None,
original_size: Tuple[int, int] | None = None,
crops_coords_top_left: Tuple[int, int] = (0, 0),
target_size: Tuple[int, int] = None,
target_size: Tuple[int, int] | None = None,
negative_original_size: Optional[Tuple[int, int]] = None,
negative_crops_coords_top_left: Tuple[int, int] = (0, 0),
negative_target_size: Optional[Tuple[int, int]] = None,