ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-04-27 17:51:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

Make 100% sure lora alpha matches for flux

Browse Source
This commit is contained in:
Jaret Burkett
2024-08-13 14:24:03 -06:00
parent 465bc1e2f8
commit f7cf2f866f
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/lora_special.py
View File

@@ -236,7 +236,9 @@ class LoRASpecialNetwork(ToolkitNetworkMixin, LoRANetwork):
if self.peft_format: if self.peft_format:
# no alpha for peft # no alpha for peft
self.alpha = self.lora_dim self.alpha = self.lora_dim
alpha = self.alpha
self.conv_alpha = self.conv_lora_dim self.conv_alpha = self.conv_lora_dim
conv_alpha = self.conv_alpha
self.full_train_in_out = full_train_in_out self.full_train_in_out = full_train_in_out