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

Composable LoRA

Browse Source
This commit is contained in:
AI-Casanova
2024-03-02 17:52:08 -06:00
parent 0d6c36ff27
commit 2da60f8628
3 changed files with 78 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/processing_diffusers.py
+9 -1
View File
@@ -8,7 +8,7 @@ import numpy as np
import torch
import torchvision.transforms.functional as TF
import diffusers
from modules import shared, devices, processing, sd_samplers, sd_models, images, errors, prompt_parser_diffusers, sd_hijack_hypertile, processing_correction, processing_vae, sd_models_compile
from modules import shared, devices, processing, sd_samplers, sd_models, images, errors, prompt_parser_diffusers, sd_hijack_hypertile, processing_correction, processing_vae, sd_models_compile, extra_networks
from modules.processing_helpers import resize_init_images, resize_hires, fix_prompts, calculate_base_steps, calculate_hires_steps, calculate_refiner_steps
from modules.onnx_impl import preprocess_pipeline as preprocess_onnx_pipeline, check_parameters_changed as olive_check_parameters_changed
@@ -73,6 +73,14 @@ def process_diffusers(p: processing.StableDiffusionProcessing):
if shared.state.interrupted or shared.state.skipped:
raise AssertionError('Interrupted...')
time.sleep(0.1)
if hasattr(p, "extra_network_data"):
if shared.opts.lora_force_diffusers:
shared.log.warning("Composable LoRA not compatible with 'lora_force_diffusers'")
else:
functional = shared.opts.lora_functional
shared.opts.lora_functional = True
extra_networks.activate(p, p.extra_network_data, step=step)
shared.opts.lora_functional = functional
if latents is None:
return kwargs
elif shared.opts.nan_skip: