From f7cf2f866fd4eb01f758f2d6a7aaa1e1738176fa Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Tue, 13 Aug 2024 14:24:03 -0600
Subject: [PATCH] Make 100% sure lora alpha matches for flux

---
 toolkit/lora_special.py | 2 ++
 1 file changed, 2 insertions(+)

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