Format code (#1154)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot] 2023-09-01 14:13:25 +08:00 committed by GitHub
parent 8ffdcb0128
commit eefb7dbe1d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 12 additions and 4 deletions

View File

@ -264,9 +264,7 @@ class Pipeline(object):
with torch.no_grad():
hasp = pitch is not None and pitchf is not None
arg = (feats, p_len, pitch, pitchf, sid) if hasp else (feats, p_len, sid)
audio1 = (
(net_g.infer(*arg)[0][0, 0]).data.cpu().float().numpy()
)
audio1 = (net_g.infer(*arg)[0][0, 0]).data.cpu().float().numpy()
del hasp, arg
del feats, p_len, padding_mask
if torch.cuda.is_available():

View File

@ -2,6 +2,7 @@ import os
from fairseq import checkpoint_utils
def get_index_path_from_model(sid):
return next(
(

View File

@ -338,5 +338,14 @@ class RVC:
.float()
)
t5 = ttime()
print("Spent time: fea =", t2 - t1, ", index =", t3 - t2, ", f0 =", t4 - t3, ", model =", t5 - t4)
print(
"Spent time: fea =",
t2 - t1,
", index =",
t3 - t2,
", f0 =",
t4 - t3,
", model =",
t5 - t4,
)
return infered_audio