diff --git a/demo2.py b/demo2.py index 18f7dec..f536429 100644 --- a/demo2.py +++ b/demo2.py @@ -3,7 +3,7 @@ from model import FunASRNano def main(): model_dir = "FunAudioLLM/Fun-ASR-Nano-2512" - m, kwargs = FunASRNano.from_pretrained(model=model_dir, device="cpu") + m, kwargs = FunASRNano.from_pretrained(model=model_dir, device="cuda:0") m.eval() wav_path = f"{kwargs['model_path']}/example/zh.mp3" diff --git a/requirements.txt b/requirements.txt index 6cef3df..596c112 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,2 +1,4 @@ +torch +torchaudio +transformers funasr>=1.2.7 -openai-whisper>=20250625