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()