fix dylora loraplus

kohya-ss · kohya-ss · commit 3fd8cdc55d7d · 2024-05-06T14:03:19.000+09:00
diff --git a/networks/dylora.py b/networks/dylora.py
@@ -466,13 +466,13 @@ def assemble_params(loras, lr, ratio):
             params = assemble_params(
                 self.text_encoder_loras,
                 text_encoder_lr if text_encoder_lr is not None else default_lr,
-                self.loraplus_text_encoder_lr_ratio or self.loraplus_ratio,
+                self.loraplus_text_encoder_lr_ratio or self.loraplus_lr_ratio,
             )
             all_params.extend(params)
 
         if self.unet_loras:
             params = assemble_params(
-                self.unet_loras, default_lr if unet_lr is None else unet_lr, self.loraplus_unet_lr_ratio or self.loraplus_ratio
+                self.unet_loras, default_lr if unet_lr is None else unet_lr, self.loraplus_unet_lr_ratio or self.loraplus_lr_ratio
             )
             all_params.extend(params)
 

Original file line number	Diff line number	Diff line change
`@@ -466,13 +466,13 @@ def assemble_params(loras, lr, ratio):`
`466`	`466`	`params = assemble_params(`
`467`	`467`	`self.text_encoder_loras,`
`468`	`468`	`text_encoder_lr if text_encoder_lr is not None else default_lr,`
`469`		`- self.loraplus_text_encoder_lr_ratio or self.loraplus_ratio,`
	`469`	`+ self.loraplus_text_encoder_lr_ratio or self.loraplus_lr_ratio,`
`470`	`470`	`)`
`471`	`471`	`all_params.extend(params)`
`472`	`472`
`473`	`473`	`if self.unet_loras:`
`474`	`474`	`params = assemble_params(`
`475`		`- self.unet_loras, default_lr if unet_lr is None else unet_lr, self.loraplus_unet_lr_ratio or self.loraplus_ratio`
	`475`	`+ self.unet_loras, default_lr if unet_lr is None else unet_lr, self.loraplus_unet_lr_ratio or self.loraplus_lr_ratio`
`476`	`476`	`)`
`477`	`477`	`all_params.extend(params)`
`478`	`478`