This commit is contained in:
ylzz1997 2023-05-25 00:52:14 +08:00
parent 68ce0b31c1
commit 3e36e2b389
1 changed files with 2 additions and 2 deletions

View File

@ -106,10 +106,10 @@ def process_one(filename, hmodel,f0p,diff=False,mel_extractor=None):
def process_batch(filenames,f0p,diff=False,mel_extractor=None):
print("Loading hubert for content...")
print("Loading speech encoder for content...")
device = "cuda" if torch.cuda.is_available() else "cpu"
hmodel = utils.get_speech_encoder(speech_encoder,device=device)
print("Loaded hubert.")
print("Loaded speech encoder.")
for filename in tqdm(filenames):
process_one(filename, hmodel,f0p,diff,mel_extractor)