fix #835: some broken imports

This commit is contained in:
源文雨 2023-07-23 14:32:53 +08:00
parent a002f817df
commit 18067aa85d
5 changed files with 6 additions and 6 deletions

View File

@ -34,7 +34,7 @@ from lib.infer_pack.models import (
from lib.infer_pack.models_onnx import SynthesizerTrnMsNSFsidM
from infer_uvr5 import _audio_pre_, _audio_pre_new
from my_utils import load_audio
from train.process_ckpt import change_info, extract_small_model, merge, show_info
from lib.train.process_ckpt import change_info, extract_small_model, merge, show_info
from vc_infer_pipeline import VC
from sklearn.cluster import MiniBatchKMeans

View File

@ -3,8 +3,8 @@ import numpy as np
import torch
import torch.utils.data
from mel_processing import spectrogram_torch
from utils import load_wav_to_torch, load_filepaths_and_text
from lib.train.mel_processing import spectrogram_torch
from lib.train.utils import load_wav_to_torch, load_filepaths_and_text
class TextAudioLoaderMultiNSFsid(torch.utils.data.Dataset):

View File

@ -1,5 +1,4 @@
import torch
from torch.nn import functional as F
def feature_loss(fmap_r, fmap_g):

View File

@ -1,4 +1,4 @@
import torch, traceback, os, pdb, sys
import torch, traceback, os, sys
now_dir = os.getcwd()
sys.path.append(now_dir)

View File

@ -44,9 +44,10 @@ def load_checkpoint_d(checkpoint_path, combd, sbd, optimizer=None, load_opt=1):
model.module.load_state_dict(new_state_dict, strict=False)
else:
model.load_state_dict(new_state_dict, strict=False)
return model
go(combd, "combd")
go(sbd, "sbd")
model = go(sbd, "sbd")
#############
logger.info("Loaded model weights")