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

2026-02-19 11:57:44 -07:00
parent 3632656cda
commit 1c74ca5d22
6 changed files with 39 additions and 10 deletions
--- a/ui/src/types.ts
+++ b/ui/src/types.ts
@@ -146,6 +146,7 @@ export interface TrainConfig {
  loss_type: 'mse' | 'mae' | 'wavelet' | 'stepped';
  do_differential_guidance?: boolean;
  differential_guidance_scale?: number;
+  audio_loss_multiplier?: number;
 }

 export interface QuantizeKwargsConfig {