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

add interrupt to diffusers

Browse Source
This commit is contained in:
Vladimir Mandic
2023-08-31 13:39:35 -04:00
parent 5856aa00e9
commit b33f3866b0
4 changed files with 20 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/lora_diffusers.py
+2 -3
View File
@@ -45,7 +45,6 @@ def load_diffusers_lora(name, lora, strength = 1.0):
lora_state['multiplier'].append(strength)
if shared.opts.diffusers_lora_loader == "diffusers default":
pipe.load_lora_weights(lora.filename, cache_dir=shared.opts.diffusers_dir, local_files_only=True, lora_scale=strength)
shared.log.info(f"LoRA loaded: {name} {lora_state['multiplier']}")
else:
from safetensors.torch import load_file
lora_sd = load_file(lora.filename)
@@ -61,9 +60,9 @@ def load_diffusers_lora(name, lora, strength = 1.0):
lora_network.to(shared.device, dtype=pipe.unet.dtype)
lora_network.apply_to(multiplier=strength)
lora_state['all_loras'].append(lora_network)
shared.log.info(f"LoRA loaded: {name}:{strength} loader={shared.opts.diffusers_lora_loader}")
shared.log.info(f"LoRA loaded: {name} strength={strength} loader={shared.opts.diffusers_lora_loader}")
except Exception as e:
shared.log.error(f"Diffusers LoRA loading failed: {name} {e}")
shared.log.error(f"LoRA loading failed: {name} {e}")
# Diffusersで動くLoRA。このファイル単独で完結する。