Fix new bug I accidently introduced with lora
This commit is contained in:
1 parent
ca3ce0f34c
commit
1e0bff653c
1 file changed
+2
@@ -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:
|
||||
|
||||
Reference in new issue
Block a user