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

Merge pull request #4935 from QualiaRain/fix/lora-network-shape-guard

Browse Source 
fix(lora): guard self.shape for modules without a weight attribute
This commit is contained in:
Vladimir Mandic
2026-06-16 09:01:24 +02:00
committed by GitHub
parent 8f5759ac01 42a4b82c8e
commit 6cc2a9582f
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/lora/network.py
+2 -1
View File
@@ -166,6 +166,7 @@ class NetworkModule:
self.network_key = weights.network_key
self.sd_key = weights.sd_key
self.sd_module = weights.sd_module
self.shape = None
if hasattr(self.sd_module, 'weight'):
if hasattr(self.sd_module, "sdnq_dequantizer"):
self.shape = self.sd_module.sdnq_dequantizer.original_shape
@@ -176,7 +177,7 @@ class NetworkModule:
self.alpha = weights.w["alpha"].item() if "alpha" in weights.w else None
self.scale = weights.w["scale"].item() if "scale" in weights.w else None
self.dora_scale = weights.w.get("dora_scale", None)
self.dora_norm_dims = len(self.shape) - 1
self.dora_norm_dims = (len(self.shape) - 1) if self.shape is not None else None
def multiplier(self):
unet_multiplier = 3 * [self.network.unet_multiplier] if not isinstance(self.network.unet_multiplier, list) else self.network.unet_multiplier