ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-05-02 04:01:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix new bug I accidently introduced with lora

Browse Source
This commit is contained in:
Jaret Burkett
2025-04-13 21:15:07 -06:00
parent ca3ce0f34c
commit 1e0bff653c
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
toolkit/lora_special.py
View File

@@ -341,6 +341,8 @@ class LoRASpecialNetwork(ToolkitNetworkMixin, LoRANetwork):
skip = True
if self.transformer_only and is_unet:
transformer_block_names = None
if base_model is not None:
transformer_block_names = base_model.get_transformer_block_names()
if transformer_block_names is not None: