diff --git a/models/Model_AMP/config_schema.json b/models/Model_AMP/config_schema.json
index 9630537..2b3079f 100644
--- a/models/Model_AMP/config_schema.json
+++ b/models/Model_AMP/config_schema.json
@@ -4,11 +4,16 @@
   "definitions": {
     "dfl_config": {
       "type": "object",
-      "additionalProperties": false,
+      "additionalProperties": true,
       "properties": {
         "use_fp16": {
           "type": "boolean"
         },
+        "cpu_cap": {
+          "type": "integer",
+          "minimum": 1,
+          "maximum": 256
+        },
         "morph_factor": {
           "type": "number",
           "minimum":0.0,
@@ -205,6 +210,7 @@
         }
       },
       "required": [
+        "cpu_cap",
         "adabelief",
         "ae_dims",
         "autobackup_hour",
@@ -232,12 +238,10 @@
         "maximum_n_backups",
         "models_opt_on_gpu",
         "mouth_prio",
-        "pretrain",
         "random_blur",
         "random_color",
         "random_downsample",
         "random_dst_flip",
-        "random_hsv_power",
         "random_jpeg",
         "random_noise",
         "random_src_flip",
diff --git a/models/Model_XSeg/config_schema.json b/models/Model_XSeg/config_schema.json
index e53fc8b..927b5a5 100644
--- a/models/Model_XSeg/config_schema.json
+++ b/models/Model_XSeg/config_schema.json
@@ -31,7 +31,7 @@
       "required": [
         "batch_size",
         "face_type",
-        "pretrain",
+        "pretrain"
       ],
       "title": "dfl_config"
     }