From 0dcbabf6afaed85e3bfba93e18a664d2af66d077 Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Wed, 1 Apr 2026 10:38:31 -0600 Subject: [PATCH] Fix merge nertwork ref --- jobs/process/BaseSDTrainProcess.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jobs/process/BaseSDTrainProcess.py b/jobs/process/BaseSDTrainProcess.py index cf7c226..0ef7077 100644 --- a/jobs/process/BaseSDTrainProcess.py +++ b/jobs/process/BaseSDTrainProcess.py @@ -632,7 +632,7 @@ class BaseSDTrainProcess(BaseTrainProcess): print_acc("Merging network weights into full model for saving...") - self.network.merge_in(merge_weight=self.train_config.merge_network_on_save_weight) + self.network.merge_in(merge_weight=self.train_config.merge_network_on_save_strength) # reset weights to zero self.network.reset_weights() self.network.is_merged_in = False