Add a merge network on save strength

This commit is contained in:
Jaret Burkett committed 2026-04-01 09:21:08 -06:00
1 parent 853ffaf207
commit da2a79590f
2 files changed
+2 -1

No files matched your search

+1 -1
View File
@@ -632,7 +632,7 @@ class BaseSDTrainProcess(BaseTrainProcess):
print_acc("Merging network weights into full model for saving...")
self.network.merge_in(merge_weight=1.0)
self.network.merge_in(merge_weight=self.train_config.merge_network_on_save_weight)
# reset weights to zero
self.network.reset_weights()
self.network.is_merged_in = False