update schedulers and lint everything

Signed-off-by: vladmandic <mandic00@live.com>
This commit is contained in:
vladmandic
2026-02-04 11:58:02 +01:00
parent f439d51ea7
commit d7ca4f63a7
46 changed files with 467 additions and 231 deletions
+3 -3
View File
@@ -44,9 +44,9 @@ class Script(scripts_manager.Script):
return None
import yaml
from scripts.ctrlx import CtrlXStableDiffusionXLPipeline
from scripts.ctrlx.sdxl import get_control_config, register_control
from scripts.ctrlx.utils import get_self_recurrence_schedule
from scripts.ctrlx import CtrlXStableDiffusionXLPipeline # pylint: disable=no-name-in-module
from scripts.ctrlx.sdxl import get_control_config, register_control # pylint: disable=no-name-in-module
from scripts.ctrlx.utils import get_self_recurrence_schedule # pylint: disable=no-name-in-module
orig_prompt_attention = shared.opts.prompt_attention
shared.opts.data['prompt_attention'] = 'fixed'
+1 -1
View File
@@ -58,7 +58,7 @@ class Script(scripts_manager.Script):
shared.log.warning('FreeScale: missing input image')
return None
from scripts.freescale import StableDiffusionXLFreeScale, StableDiffusionXLFreeScaleImg2Img
from scripts.freescale import StableDiffusionXLFreeScale, StableDiffusionXLFreeScaleImg2Img # pylint: disable=no-name-in-module
self.orig_pipe = shared.sd_model
self.orig_slice = shared.opts.diffusers_vae_slicing
self.orig_tile = shared.opts.diffusers_vae_tiling
+1 -1
View File
@@ -19,7 +19,7 @@ def verify_insightface():
def load_infiniteyou(model: str):
from scripts.infiniteyou import InfUFluxPipeline
from scripts.infiniteyou import InfUFluxPipeline # pylint: disable=no-name-in-module
shared.sd_model = InfUFluxPipeline(
pipe=shared.sd_model,
model_version=model,
+1 -1
View File
@@ -1,5 +1,5 @@
from safetensors.torch import load_file
from scripts.layerdiffuse.layerdiffuse_model import LoraLoader, AttentionSharingProcessor
from scripts.layerdiffuse.layerdiffuse_model import LoraLoader, AttentionSharingProcessor # pylint: disable=no-name-in-module
def merge_delta_weights_into_unet(pipe, delta_weights):
+2 -2
View File
@@ -65,7 +65,7 @@ class Script(scripts_manager.Script):
if repo_id is not None:
import huggingface_hub as hf
repo_file = hf.snapshot_download(repo_id, cache_dir=shared.opts.hfcache_dir)
from scripts.lbm import get_model
from scripts.lbm import get_model # pylint: disable=no-name-in-module
model = get_model(
repo_file,
save_dir=None,
@@ -85,7 +85,7 @@ class Script(scripts_manager.Script):
install('lpips')
from torchvision.transforms import ToPILImage, ToTensor
from scripts.lbm import get_model, extract_object, resize_and_center_crop
from scripts.lbm import get_model, extract_object, resize_and_center_crop # pylint: disable=no-name-in-module
ori_h_bg, ori_w_bg = fg_image.size
ar_bg = ori_h_bg / ori_w_bg
+1 -1
View File
@@ -49,7 +49,7 @@ class Script(scripts_manager.Script):
supported_model_list = ['sdxl']
if shared.sd_model_type not in supported_model_list:
shared.log.warning(f'PixelSmith: class={shared.sd_model.__class__.__name__} model={shared.sd_model_type} required={supported_model_list}')
from scripts.pixelsmith import PixelSmithXLPipeline, PixelSmithVAE
from scripts.pixelsmith import PixelSmithXLPipeline, PixelSmithVAE # pylint: disable=no-name-in-module
self.orig_pipe = shared.sd_model
self.orig_vae = shared.sd_model.vae
if self.vae is None:
+1 -1
View File
@@ -123,7 +123,7 @@ class StableDiffusionXLPuLIDPipeline:
if sampler is not None:
self.sampler = sampler
else:
from scripts.pulid import sampling
from scripts.pulid import sampling # pylint: disable=no-name-in-module
self.sampler = sampling.sample_dpmpp_sde
@property
+4 -4
View File
@@ -34,11 +34,11 @@ class Script(scripts_manager.Script):
return model, sampler, width, height, start, scale, lora
def run(self, p: processing.StableDiffusionProcessing, model, sampler, width, height, start, scale, lora): # pylint: disable=arguments-differ, unused-argument
from scripts.xadapter.xadapter_hijacks import PositionNet
from scripts.xadapter.xadapter_hijacks import PositionNet # pylint: disable=no-name-in-module
diffusers.models.embeddings.PositionNet = PositionNet # patch diffusers==0.26 from diffusers==0.20
from scripts.xadapter.adapter import Adapter_XL
from scripts.xadapter.pipeline_sd_xl_adapter import StableDiffusionXLAdapterPipeline
from scripts.xadapter.unet_adapter import UNet2DConditionModel as UNet2DConditionModelAdapter
from scripts.xadapter.adapter import Adapter_XL # pylint: disable=no-name-in-module
from scripts.xadapter.pipeline_sd_xl_adapter import StableDiffusionXLAdapterPipeline # pylint: disable=no-name-in-module
from scripts.xadapter.unet_adapter import UNet2DConditionModel as UNet2DConditionModelAdapter # pylint: disable=no-name-in-module
global adapter # pylint: disable=global-statement
if model == 'None':