From 7f551b6787249c801963c7a5565292f0f769cad0 Mon Sep 17 00:00:00 2001
From: Ftps <ftpsflandre@gmail.com>
Date: Sun, 19 Nov 2023 04:12:05 +0900
Subject: [PATCH] fix time

---
 rvc/modules/vc/pipeline.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/rvc/modules/vc/pipeline.py b/rvc/modules/vc/pipeline.py
index 88f2166..2b4437c 100644
--- a/rvc/modules/vc/pipeline.py
+++ b/rvc/modules/vc/pipeline.py
@@ -274,8 +274,8 @@ class Pipeline(object):
         if torch.cuda.is_available():
             torch.cuda.empty_cache()
         t2 = ttime()
-        times[0] += t1 - t0
-        times[2] += t2 - t1
+        times["npy"] += t1 - t0
+        times["infer"] += t2 - t1
         return audio1
 
     def pipeline(
@@ -367,7 +367,7 @@ class Pipeline(object):
             pitch = torch.tensor(pitch, device=self.device).unsqueeze(0).long()
             pitchf = torch.tensor(pitchf, device=self.device).unsqueeze(0).float()
         t2 = ttime()
-        times[1] += t2 - t1
+        times["f0"] += t2 - t1
         for t in opt_ts:
             t = t // self.window * self.window
             if if_f0 == 1: