From 4909b809c70ab79ac80c285e67fa8d4a59adc03b Mon Sep 17 00:00:00 2001 From: Jaret Burkett Date: Tue, 10 Mar 2026 15:16:09 -0600 Subject: [PATCH] Fixed issue with audio loss multiplier. --- toolkit/config_modules.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/toolkit/config_modules.py b/toolkit/config_modules.py index 03c04cf1..418d635d 100644 --- a/toolkit/config_modules.py +++ b/toolkit/config_modules.py @@ -567,6 +567,8 @@ class TrainConfig: # stabilizes empty prompts to be zeroed predictions self.do_blank_stabilization = kwargs.get('do_blank_stabilization', False) + + self.audio_loss_multiplier = kwargs.get("audio_loss_multiplier", 1.0) ModelArch = Literal['sd1', 'sd2', 'sd3', 'sdxl', 'pixart', 'pixart_sigma', 'auraflow', 'flux', 'flex1', 'flex2', 'lumina2', 'vega', 'ssd', 'wan21'] @@ -687,8 +689,6 @@ class ModelConfig: # model paths for models that support it self.model_paths = kwargs.get("model_paths", {}) - self.audio_loss_multiplier = kwargs.get("audio_loss_multiplier", 1.0) - # allow frontend to pass arch with a color like arch:tag # but remove the tag if self.arch is not None: