ostris/ai-toolkit
1
0
Fork 0
mirror of https://github.com/ostris/ai-toolkit.git synced 2026-03-12 22:19:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed issue with audio loss multiplier.

Browse Source
This commit is contained in:
Jaret Burkett
2026-03-10 15:16:09 -06:00
parent 06ef3d343a
commit 4909b809c7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
toolkit/config_modules.py
View File

@@ -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: