monster/Qwen-Image-Layered-MRP-MLX
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix error in lora_layers causing lora training to fail

Browse Source
This commit is contained in:
filipstrand 2025-06-12 23:36:00 +02:00
parent 4eb3f90fc9
commit d2296ec197
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/mflux/dreambooth/lora_layers/lora_layers.py
View File

@ -246,7 +246,8 @@ class LoRALayers:
if block[key].get(name, False) and val.get(name, False): if block[key].get(name, False) and val.get(name, False):
if name == "to_out" and isinstance(block[key][name], list): if name == "to_out" and isinstance(block[key][name], list):
if len(block[key][name]) > 0: if len(block[key][name]) > 0:
block[key][name][0] = val[name] lora_layer = val[name][0] if isinstance(val[name], list) else val[name]
block[key][name][0] = lora_layer
else: else:
block[key][name] = val[name] block[key][name] = val[name]