fix #615

mikazuki/schema/flux-lora.ts

@@ -73,6 +73,9 @@ Schema.intersect([
     // FLUX 去除 max_token_length
     Schema.object(UpdateSchema(SHARED_SCHEMAS.RAW.CAPTION_SETTINGS, {}, ["max_token_length"])).description("caption（Tag）选项"),
 
+    // 噪声设置
+    SHARED_SCHEMAS.NOISE_SETTINGS,
+
     // 数据增强
     SHARED_SCHEMAS.DATA_ENCHANCEMENT,
 

mikazuki/schema/lora-master.ts

@@ -79,6 +79,9 @@ Schema.intersect([
     // caption 选项
     Schema.object(SHARED_SCHEMAS.RAW.CAPTION_SETTINGS).description("caption（Tag）选项"),
 
+    // 噪声设置
+    SHARED_SCHEMAS.NOISE_SETTINGS,
+
     // 数据增强
     SHARED_SCHEMAS.DATA_ENCHANCEMENT,
 

mikazuki/schema/shared.ts

@@ -108,6 +108,7 @@
                 Schema.object({})
             ])
         ]).description("保存设置"),
+
         LR_OPTIMIZER: Schema.intersect([
             Schema.object({
                 learning_rate: Schema.string().default("1e-4").description("总学习率, 在分开设置 U-Net 与文本编码器学习率后这个值失效。"),
@@ -208,6 +209,12 @@
             ]),
         ]),
 
+        NOICE_SETTINGS: Schema.object({
+            noise_offset: Schema.number().step(0.01).description("在训练中添加噪声偏移来改良生成非常暗或者非常亮的图像，如果启用推荐为 0.1"),
+            multires_noise_iterations: Schema.number().step(1).description("多分辨率（金字塔）噪声迭代次数 推荐 6-10。无法与 noise_offset 一同启用"),
+            multires_noise_discount: Schema.number().step(0.01).description("多分辨率（金字塔）衰减率 推荐 0.3-0.8，须同时与上方参数 multires_noise_iterations 一同启用"),
+        }).description("噪声设置"),
+
         DATA_ENCHANCEMENT: Schema.object({
             color_aug: Schema.boolean().description("颜色改变"),
             flip_aug: Schema.boolean().description("图像翻转"),