浏览代码

Optimize path handling.

Lengyue 1 年之前
父节点
当前提交
ca2fd76d83
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      tools/vqgan/inference.py

+ 1 - 1
tools/vqgan/inference.py

@@ -69,7 +69,7 @@ def main(input_path, output_path, config_name, checkpoint_path, device):
         logger.info(f"Processing in-place reconstruction of {input_path}")
 
         # Load audio
-        audio, sr = torchaudio.load(input_path)
+        audio, sr = torchaudio.load(str(input_path))
         if audio.shape[0] > 1:
             audio = audio.mean(0, keepdim=True)
         audio = torchaudio.functional.resample(audio, sr, model.sampling_rate)