add vae_batch_size

Akegarasu · Aug 31, 2024 · 9aace3e · 9aace3e
1 parent 931392e
commit 9aace3e
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 0 deletions.
diff --git a/mikazuki/schema/dreambooth-master.ts b/mikazuki/schema/dreambooth-master.ts
@@ -200,6 +200,7 @@ Schema.intersect([
         cache_latents: Schema.boolean().default(true).description("缓存图像 latent"),
         cache_latents_to_disk: Schema.boolean().default(true).description("缓存图像 latent 到磁盘"),
         persistent_data_loader_workers: Schema.boolean().default(true).description("保留加载训练集的worker，减少每个 epoch 之间的停顿。"),
+        vae_batch_size: Schema.number().min(1).description("vae 编码批量大小"),
     }).description("速度优化选项"),
 
     Schema.object({

diff --git a/mikazuki/schema/lora-flux-master.ts b/mikazuki/schema/lora-flux-master.ts
@@ -202,6 +202,7 @@ Schema.intersect([
         cache_text_encoder_outputs: Schema.boolean().default(true).description("缓存文本编码器的输出，减少显存使用。使用时需要关闭 shuffle_caption"),
         cache_text_encoder_outputs_to_disk: Schema.boolean().default(true).description("缓存文本编码器的输出到磁盘"),
         persistent_data_loader_workers: Schema.boolean().default(true).description("保留加载训练集的worker，减少每个 epoch 之间的停顿。"),
+        vae_batch_size: Schema.number().min(1).description("vae 编码批量大小"),
     }).description("速度优化选项"),
 
     Schema.object({

diff --git a/mikazuki/schema/lora-master.ts b/mikazuki/schema/lora-master.ts
@@ -250,6 +250,7 @@ Schema.intersect([
         cache_text_encoder_outputs: Schema.boolean().description("缓存文本编码器的输出，减少显存使用。使用时需要关闭 shuffle_caption"),
         cache_text_encoder_outputs_to_disk: Schema.boolean().description("缓存文本编码器的输出到磁盘"),
         persistent_data_loader_workers: Schema.boolean().default(true).description("保留加载训练集的worker，减少每个 epoch 之间的停顿。"),
+        vae_batch_size: Schema.number().min(1).description("vae 编码批量大小"),
     }).description("速度优化选项"),
 
     Schema.object({