mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 01:04:32 +02:00
update schedulers and lint everything
Signed-off-by: vladmandic <mandic00@live.com>
This commit is contained in:
@@ -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'
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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,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
@@ -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
|
||||
|
||||
@@ -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:
|
||||
|
||||
@@ -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
|
||||
|
||||
@@ -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':
|
||||
|
||||
Reference in New Issue
Block a user