Skip to content

Commit 9ee9f5f

Browse files
committed
fix: Pipeline.pipeline for empty f0_file
- Added checks for empty f0_file, skipping faulty inp_f0 creation if empty as it should
1 parent 96e3d8a commit 9ee9f5f

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

infer/modules/vc/pipeline.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -324,11 +324,13 @@ def pipeline(
324324
if hasattr(f0_file, "name"):
325325
try:
326326
with open(f0_file.name, "r") as f:
327-
lines = f.read().strip("\n").split("\n")
328-
inp_f0 = []
329-
for line in lines:
330-
inp_f0.append([float(i) for i in line.split(",")])
331-
inp_f0 = np.array(inp_f0, dtype="float32")
327+
raw_lines = f.read()
328+
if len(raw_lines) > 0:
329+
lines = raw_lines.strip("\n").split("\n")
330+
inp_f0 = []
331+
for line in lines:
332+
inp_f0.append([float(i) for i in line.split(",")])
333+
inp_f0 = np.array(inp_f0, dtype="float32")
332334
except:
333335
traceback.print_exc()
334336
sid = torch.tensor(sid, device=self.device).unsqueeze(0).long()

0 commit comments

Comments
 (0)