Allow control image for omnigen training and sampling

2025-07-09 13:54:55 -06:00
parent bbb57de6ec
commit 611969ec1f
6 changed files with 187 additions and 132 deletions
--- a/ui/src/types.ts
+++ b/ui/src/types.ts
@@ -110,6 +110,8 @@ export interface TrainConfig {
  optimizer_params: {
    weight_decay: number;
  };
+  skip_first_sample: boolean;
+  disable_sampling: boolean;
  diff_output_preservation: boolean;
  diff_output_preservation_multiplier: number;
  diff_output_preservation_class: string;