github/ai-toolkit
1
0
Fork 0
Code Issues Pull Requests Actions 1 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

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():