From fb56d561aa0c1591331d029b2358cb0ac48f9102 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Kai=20Li=20=28=E6=9D=8E=E5=87=AF=29?=
 <tsinghua.kaili@gmail.com>
Date: Wed, 18 Sep 2024 09:23:47 +0800
Subject: [PATCH] Update inference.py

---
 inference.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/inference.py b/inference.py
index ce97d90..872c77c 100644
--- a/inference.py
+++ b/inference.py
@@ -15,7 +15,7 @@ def save_audio(file_path, audio, samplerate=44100):
 def main(input_wav, output_wav):
     os.environ['CUDA_VISIBLE_DEVICES'] = "0"
 
-    model = look2hear.models.BaseModel.from_pretrain("JusperLee/Apollo").cuda()
+    model = look2hear.models.BaseModel.from_pretrain("JusperLee/Apollo", sr=44100, win=20, feature_dim=256, layer=6).cuda()
     test_data = load_audio(input_wav)
     with torch.no_grad():
         out = model(test_data)