Add model compiling to the ui

2026-06-12 11:02:10 -06:00
parent 6b1f89f30b
commit c09b228a35
4 changed files with 37 additions and 3 deletions
--- a/ui/src/types.ts
+++ b/ui/src/types.ts
@@ -151,6 +151,12 @@ export interface TrainConfig {
  differential_guidance_scale?: number;
  audio_loss_multiplier?: number;
  max_loss?: number | null;
+  compile?: boolean;
+  block_compile?: boolean;
+  compile_mode?: 'default' | 'max-autotune' | 'fastest';
+  compile_fullgraph?: boolean;
+  compile_dynamic?: boolean;
+  cache_size_limit?: number;
 }

 export interface QuantizeKwargsConfig {