Added Differential Guidance training target

2025-11-10 09:38:25 -07:00
parent 9b89bab8fe
commit 2e7b2d9926
9 changed files with 101 additions and 6 deletions
--- a/ui/src/types.ts
+++ b/ui/src/types.ts
@@ -139,6 +139,8 @@ export interface TrainConfig {
  blank_prompt_preservation_multiplier?: number;
  switch_boundary_every: number;
  loss_type: 'mse' | 'mae' | 'wavelet' | 'stepped';
+  do_differential_guidance?: boolean;
+  differential_guidance_scale?: number;
 }

 export interface QuantizeKwargsConfig {