ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-01-26 16:39:47 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed misnamed var

Browse Source
This commit is contained in:
Jaret Burkett
2024-07-20 23:00:20 +00:00
parent c2c4e8cf34
commit c51235c486
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/models/ilora2.py
View File

@@ -138,7 +138,7 @@ class InstantLoRAMidModule(torch.nn.Module):
self.down_dim = self.down_shape[1] if self.do_down else 0
self.mid_dim = self.up_shape[1] if self.do_mid else 0
self.out_dim = self.up_shape[0] if self.do_down else 0
self.out_dim = self.up_shape[0] if self.do_up else 0
self.embed = None