From 0812020c900f7d1f5089d81d52086aa34c82ba4d Mon Sep 17 00:00:00 2001
From: RVC-Boss <129054828+RVC-Boss@users.noreply.github.com>
Date: Sun, 18 Jun 2023 19:37:53 +0800
Subject: [PATCH] Add files via upload

---
 train/utils.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/train/utils.py b/train/utils.py
index 434ed7a..984c8c1 100644
--- a/train/utils.py
+++ b/train/utils.py
@@ -360,7 +360,10 @@ def get_hparams(init=True):
     if not os.path.exists(experiment_dir):
         os.makedirs(experiment_dir)
 
-    config_path = "configs/%s.json" % args.sample_rate
+    if(args.version=="v1"or args.sample_rate=="40k"):
+        config_path = "configs/%s.json" % args.sample_rate
+    else:
+        config_path = "configs/%s_v2.json" % args.sample_rate
     config_save_path = os.path.join(experiment_dir, "config.json")
     if init:
         with open(config_path, "r") as f: