Revert "func -> call"

This reverts commit 043fa4d750.
2024-06-02 02:34:38 +09:00
parent 790b504ba2
commit e02e00753d
6 changed files with 18 additions and 50 deletions
@@ -4,7 +4,6 @@ from pathlib import Path
 import click
 from dotenv import load_dotenv
 from scipy.io import wavfile
-import torch

 logging.getLogger("numba").setLevel(logging.WARNING)

@@ -130,6 +129,4 @@ def infer(
        wavfile.write(outputpath, tgt_sr, audio_opt)
        click.echo(times)
        click.echo(f"Finish inference. Check {outputpath}")
-
-    if torch.cuda.is_available():
-        torch.cuda.empty_cache()
+    return tgt_sr, audio_opt, times