From 4b00b6125729e5c68378c37d6a7e9fd29a0d82dd Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Wed, 12 Aug 2026 18:58:54 -0600 Subject: [PATCH] When doing an audio loss, show the img and audio loss in the loss log --- extensions_built_in/sd_trainer/SDTrainer.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/extensions_built_in/sd_trainer/SDTrainer.py b/extensions_built_in/sd_trainer/SDTrainer.py index c514ec33..c1562c31 100644 --- a/extensions_built_in/sd_trainer/SDTrainer.py +++ b/extensions_built_in/sd_trainer/SDTrainer.py @@ -994,6 +994,8 @@ class SDTrainer(BaseSDTrainProcess): if batch.audio_pred is not None and batch.audio_target is not None: audio_loss = torch.nn.functional.mse_loss(batch.audio_pred.float(), batch.audio_target.float(), reduction="mean") audio_loss = audio_loss * self.train_config.audio_loss_multiplier + self.additional_logs['loss/img'] = loss.item() + self.additional_logs['loss/audio'] = audio_loss.item() loss = loss + audio_loss # check for additional losses