Added validation loss

ui/src/types.ts

@@ -117,6 +117,18 @@ export interface EMAConfig {
   ema_decay: number;
 }
 
+export interface ValidationItem {
+  image_path: string;
+  prompt: string;
+}
+
+export interface ValidationConfig {
+  validation_items: ValidationItem[];
+  resolution: number;
+  validate_every_n_steps: number;
+  validation_sigmas?: number[];
+}
+
 export interface TrainConfig {
   batch_size: number;
   bypass_guidance_embedding?: boolean;
@@ -151,6 +163,7 @@ export interface TrainConfig {
   differential_guidance_scale?: number;
   audio_loss_multiplier?: number;
   max_loss?: number | null;
+  validation_config?: ValidationConfig;
 }
 
 export interface QuantizeKwargsConfig {