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

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

No files matched your search

+2
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: