From ed96cf17211e8198c8115b772b16a8ac10e46a3c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=BA=90=E6=96=87=E9=9B=A8?=
 <41315874+fumiama@users.noreply.github.com>
Date: Tue, 4 Jun 2024 01:58:26 +0900
Subject: [PATCH] fix(train): RuntimeError: context has already been set

---
 infer/lib/train/utils.py     | 2 --
 infer/modules/train/train.py | 2 +-
 infer/modules/vc/pipeline.py | 3 +--
 3 files changed, 2 insertions(+), 5 deletions(-)

diff --git a/infer/lib/train/utils.py b/infer/lib/train/utils.py
index d16756a..ea7467d 100644
--- a/infer/lib/train/utils.py
+++ b/infer/lib/train/utils.py
@@ -3,9 +3,7 @@ import glob
 import json
 import logging
 import os
-import subprocess
 import sys
-import shutil
 
 import numpy as np
 import torch
diff --git a/infer/modules/train/train.py b/infer/modules/train/train.py
index 48c1f57..10afe7e 100644
--- a/infer/modules/train/train.py
+++ b/infer/modules/train/train.py
@@ -636,5 +636,5 @@ def train_and_evaluate(
 
 
 if __name__ == "__main__":
-    torch.multiprocessing.set_start_method("spawn")
+    mp.set_start_method("spawn", force=True)
     main()
diff --git a/infer/modules/vc/pipeline.py b/infer/modules/vc/pipeline.py
index 6833bd9..266b7c3 100644
--- a/infer/modules/vc/pipeline.py
+++ b/infer/modules/vc/pipeline.py
@@ -8,7 +8,6 @@ logger = logging.getLogger(__name__)
 from functools import lru_cache
 from time import time
 
-import faiss
 import librosa
 import numpy as np
 import parselmouth
@@ -331,9 +330,9 @@ class Pipeline(object):
             and os.path.exists(file_index)
             and index_rate != 0
         ):
+            if "faiss" not in sys.modules: import faiss
             try:
                 index = faiss.read_index(file_index)
-                # big_npy = np.load(file_big_npy)
                 big_npy = index.reconstruct_n(0, index.ntotal)
             except:
                 traceback.print_exc()