Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Update lora_diffusers.py

Browse Source
This commit is contained in:
AI-Casanova
2023-07-28 18:00:44 -05:00
committed by GitHub
parent 31e1bb01ff
commit 942371b25a
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/lora_diffusers.py
+2 -1
View File
@@ -10,6 +10,7 @@ lora_state = { # TODO Lora state for Diffusers
def unload_diffusers_lora():
try:
pipe = shared.sd_model
pipe.unload_lora_weights()
lora_state['active'] = False
lora_state['loaded'] = 0
pipe._remove_text_encoder_monkey_patch() # pylint: disable=W0212
@@ -24,7 +25,7 @@ def unload_diffusers_lora():
def load_diffusers_lora(name, lora, strength = 1.0):
try:
pipe = shared.sd_model
pipe.load_lora_weights(lora.filename, cache_dir=shared.opts.diffusers_dir, local_files_only=True)
pipe.load_lora_weights(lora.filename, cache_dir=shared.opts.diffusers_dir, local_files_only=True, lora_scale=strength)
lora_state['active'] = True
lora_state['loaded'] += 1
lora_state['multiplier'] = strength