소스 검색

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)