diff --git a/mikazuki/schema/flux-lora.ts b/mikazuki/schema/flux-lora.ts index 2f37e5a..9427740 100644 --- a/mikazuki/schema/flux-lora.ts +++ b/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, diff --git a/mikazuki/schema/lora-master.ts b/mikazuki/schema/lora-master.ts index 377afac..50e731a 100644 --- a/mikazuki/schema/lora-master.ts +++ b/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, diff --git a/mikazuki/schema/shared.ts b/mikazuki/schema/shared.ts index d18e215..7a921b0 100644 --- a/mikazuki/schema/shared.ts +++ b/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("图像翻转"),