diff --git a/mikazuki/schema/shared.ts b/mikazuki/schema/shared.ts
index bc719f2..ba3d441 100644
--- a/mikazuki/schema/shared.ts
+++ b/mikazuki/schema/shared.ts
@@ -99,7 +99,7 @@
                 save_precision: Schema.union(["fp16", "float", "bf16"]).default("fp16").description("模型保存精度"),
                 save_every_n_epochs: Schema.number().default(2).description("每 N epoch（轮）自动保存一次模型"),
                 save_state: Schema.boolean().default(false).description("保存训练状态 配合 `resume` 参数可以继续从某个状态训练"),
-            }),
+            }).description("保存设置"),
             Schema.union([
                 Schema.object({
                     save_state: Schema.const(true).required(),
@@ -107,7 +107,7 @@
                 }),
                 Schema.object({})
             ])
-        ]).description("保存设置"),
+        ]),
 
         LR_OPTIMIZER: Schema.intersect([
             Schema.object({