diff --git a/toolkit/lora_special.py b/toolkit/lora_special.py index df1e807b..b407618a 100644 --- a/toolkit/lora_special.py +++ b/toolkit/lora_special.py @@ -236,7 +236,9 @@ class LoRASpecialNetwork(ToolkitNetworkMixin, LoRANetwork): if self.peft_format: # no alpha for peft self.alpha = self.lora_dim + alpha = self.alpha self.conv_alpha = self.conv_lora_dim + conv_alpha = self.conv_alpha self.full_train_in_out = full_train_in_out