mirror of
https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git
synced 2026-06-06 01:30:24 +08:00
fix(dml): train extract_f0_print error
ModuleNotFoundError: No module named 'torch.privateuseone' due to new prosess
This commit is contained in:
@@ -2,6 +2,7 @@ import os
|
||||
import sys
|
||||
import traceback
|
||||
from pathlib import Path
|
||||
import importlib.util
|
||||
|
||||
from dotenv import load_dotenv
|
||||
|
||||
@@ -38,6 +39,8 @@ f0method = sys.argv[3]
|
||||
device = sys.argv[4]
|
||||
is_half = sys.argv[5] == "True"
|
||||
|
||||
if importlib.util.find_spec("torch_directml") is not None:
|
||||
import torch_directml # use side effect
|
||||
|
||||
class FeatureInput(object):
|
||||
def __init__(self, is_half: bool, device="cpu", samplerate=16000, hop_size=160):
|
||||
|
||||
Reference in New Issue
Block a user