diff --git a/mikazuki/schema/dreambooth-master.ts b/mikazuki/schema/dreambooth-master.ts
index 339b451..dff6135 100644
--- 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
index 50eb21c..45d1254 100644
--- 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
index 8d86ef3..65f5c8a 100644
--- 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({