Merge pull request #647 from ShiromiyaG/fix-my-mistakes-5

Fix some issues
IAHispano · Aug 28, 2024 · ce0857c · ce0857c
2 parents 5471dab + ca4ffcc
commit ce0857c
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/rvc/train/process/extract_model.py b/rvc/train/process/extract_model.py
@@ -57,7 +57,7 @@ def extract_model(
         else:
             dataset_lenght = None
 
-        with open(os.path.join(now_dir, "assets", "config.json"), "w") as f:
+        with open(os.path.join(now_dir, "assets", "config.json"), "r") as f:
             data = json.load(f)
             model_author = data.get("model_author", None)
 

diff --git a/rvc/train/train.py b/rvc/train/train.py
@@ -1018,10 +1018,11 @@ def save_to_json(
         )
 
         pid_file_path = os.path.join(experiment_dir, "config.json")
-        with open(pid_file_path, "w") as f:
-            pid_data = json.load(f)
+        with open(pid_file_path, "r") as pid_file:
+            pid_data = json.load(pid_file)
+        with open(pid_file_path, "w") as pid_file:
             pid_data.pop("process_pids", None)
-            json.dump(pid_data, f, indent=4)
+            json.dump(pid_data, pid_file, indent=4)
 
         if not os.path.exists(
             os.path.join(experiment_dir, f"{model_name}_{epoch}e_{global_step}s.pth")