diff --git a/train/process_ckpt.py b/train/process_ckpt.py
index ac608ae..1535d27 100644
--- a/train/process_ckpt.py
+++ b/train/process_ckpt.py
@@ -92,47 +92,91 @@ def extract_small_model(path, name, sr, if_f0, info, version):
                 40000,
             ]
         elif sr == "48k":
-            opt["config"] = [
-                1025,
-                32,
-                192,
-                192,
-                768,
-                2,
-                6,
-                3,
-                0,
-                "1",
-                [3, 7, 11],
-                [[1, 3, 5], [1, 3, 5], [1, 3, 5]],
-                [10, 6, 2, 2, 2],
-                512,
-                [16, 16, 4, 4, 4],
-                109,
-                256,
-                48000,
-            ]
+            if(version=="v1"):
+                opt["config"] = [
+                    1025,
+                    32,
+                    192,
+                    192,
+                    768,
+                    2,
+                    6,
+                    3,
+                    0,
+                    "1",
+                    [3, 7, 11],
+                    [[1, 3, 5], [1, 3, 5], [1, 3, 5]],
+                    [10, 6, 2, 2, 2],
+                    512,
+                    [16, 16, 4, 4, 4],
+                    109,
+                    256,
+                    48000,
+                ]
+            else:
+                opt["config"] = [
+                    1025,
+                    32,
+                    192,
+                    192,
+                    768,
+                    2,
+                    6,
+                    3,
+                    0,
+                    "1",
+                    [3, 7, 11],
+                    [[1, 3, 5], [1, 3, 5], [1, 3, 5]],
+                    [12,10,2,2],
+                    512,
+                    [24,20,4,4],
+                    109,
+                    256,
+                    48000,
+                ]
         elif sr == "32k":
-            opt["config"] = [
-                513,
-                32,
-                192,
-                192,
-                768,
-                2,
-                6,
-                3,
-                0,
-                "1",
-                [3, 7, 11],
-                [[1, 3, 5], [1, 3, 5], [1, 3, 5]],
-                [10, 4, 2, 2, 2],
-                512,
-                [16, 16, 4, 4, 4],
-                109,
-                256,
-                32000,
-            ]
+            if(version=="v1"):
+                opt["config"] = [
+                    513,
+                    32,
+                    192,
+                    192,
+                    768,
+                    2,
+                    6,
+                    3,
+                    0,
+                    "1",
+                    [3, 7, 11],
+                    [[1, 3, 5], [1, 3, 5], [1, 3, 5]],
+                    [10, 4, 2, 2, 2],
+                    512,
+                    [16, 16, 4, 4, 4],
+                    109,
+                    256,
+                    32000,
+                ]
+            else:
+                opt["config"] = [
+                    513,
+                    32,
+                    192,
+                    192,
+                    768,
+                    2,
+                    6,
+                    3,
+                    0,
+                    "1",
+                    [3, 7, 11],
+                    [[1, 3, 5], [1, 3, 5], [1, 3, 5]],
+                    [10,8,2,2],
+                    512,
+                    [20,16,4,4],
+                    109,
+                    256,
+                    32000,
+                ]
         if info == "":
             info = "Extracted model."
         opt["info"] = info