Add audio_loss_multiplier to scale audio loss to larger values if desired.

This commit is contained in:
Jaret Burkett
2026-02-19 11:57:44 -07:00
parent 3632656cda
commit 1c74ca5d22
6 changed files with 39 additions and 10 deletions

View File

@@ -683,6 +683,8 @@ 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: