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 bug that prevented using schnell training adapter

Browse Source
This commit is contained in:
Jaret Burkett
2025-03-19 10:25:12 -06:00
parent b829983b16
commit bbfd6ef0fe
2 changed files with 7 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
toolkit/lora_special.py
View File

@@ -209,7 +209,9 @@ class LoRASpecialNetwork(ToolkitNetworkMixin, LoRANetwork):
ignore_if_contains = []
self.ignore_if_contains = ignore_if_contains
self.transformer_only = transformer_only
self.base_model_ref = weakref.ref(base_model)
self.base_model_ref = None
if base_model is not None:
self.base_model_ref = weakref.ref(base_model)
self.only_if_contains: Union[List, None] = only_if_contains

6
toolkit/network_mixins.py
View File

@@ -547,7 +547,8 @@ class ToolkitNetworkMixin:
save_dict = new_save_dict
save_dict = self.base_model_ref().convert_lora_weights_before_save(save_dict)
if self.base_model_ref is not None:
save_dict = self.base_model_ref().convert_lora_weights_before_save(save_dict)
return save_dict
def save_weights(
@@ -586,7 +587,8 @@ class ToolkitNetworkMixin:
# probably a state dict
weights_sd = file
weights_sd = self.base_model_ref().convert_lora_weights_before_load(weights_sd)
if self.base_model_ref is not None:
weights_sd = self.base_model_ref().convert_lora_weights_before_load(weights_sd)
load_sd = OrderedDict()
for key, value in weights_sd.items():