ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-03-05 10:39:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Check for contains only against clean name for lora, not the adjusted one

Browse Source
This commit is contained in:
Jaret Burkett
2024-08-31 07:44:13 -06:00
parent 562405923f
commit 3e71a99df0
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
toolkit/lora_special.py
View File

@@ -305,15 +305,15 @@ class LoRASpecialNetwork(ToolkitNetworkMixin, LoRANetwork):
lora_name = ".".join(lora_name)
# if it doesnt have a name, it wil have two dots
lora_name.replace("..", ".")
clean_name = lora_name
if self.peft_format:
# we replace this on saving
lora_name = lora_name.replace(".", "$$")
else:
lora_name = lora_name.replace(".", "_")
skip = False
if any([word in child_name for word in self.ignore_if_contains]):
if any([word in clean_name for word in self.ignore_if_contains]):
skip = True
# see if it is over threshold
@@ -329,7 +329,7 @@ class LoRASpecialNetwork(ToolkitNetworkMixin, LoRANetwork):
if (is_linear or is_conv2d) and not skip:
if self.only_if_contains is not None and not any([word in lora_name for word in self.only_if_contains]):
if self.only_if_contains is not None and not any([word in clean_name for word in self.only_if_contains]):
continue
dim = None