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

Linting fix.

Browse Source
This commit is contained in:
Hameer Abbasi
2023-08-05 18:35:18 +00:00
parent 5a3d92883d
commit 2172f7c3f0
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/lora_diffusers.py
+4 -4
View File
@@ -32,7 +32,7 @@ def unload_diffusers_lora():
lora_state['loaded'] = 0
lora_state['all_loras'] = []
lora_state['multiplier'] = []
except Exception as e:
shared.log.error(f"Diffusers LoRA unloading failed: {e}")
@@ -355,7 +355,7 @@ class LoRANetwork(torch.nn.Module):
super().__init__()
self.multiplier = multiplier
shared.log.debug(f"create LoRA network from weights")
shared.log.debug("create LoRA network from weights")
# convert SDXL Stability AI's U-Net modules to Diffusers
converted = self.convert_unet_modules(modules_dim, modules_alpha)
@@ -496,13 +496,13 @@ class LoRANetwork(torch.nn.Module):
shared.log.debug("merge LoRA weights to original weights")
for lora in tqdm(self.text_encoder_loras + self.unet_loras):
lora.merge_to(multiplier)
shared.log.debug(f"weights are merged")
shared.log.debug("weights are merged")
def restore_from(self, multiplier=1.0):
shared.log.debug("restore LoRA weights from original weights")
for lora in tqdm(self.text_encoder_loras + self.unet_loras):
lora.restore_from(multiplier)
shared.log.debug(f"weights are restored")
shared.log.debug("weights are restored")
def load_state_dict(self, state_dict: Mapping[str, Any], strict: bool = True):
# convert SDXL Stability AI's state dict to Diffusers' based state dict