From f7b0a325c13540ebab00b8e753b6f18a77d5c771 Mon Sep 17 00:00:00 2001 From: jingyaogong Date: Mon, 11 May 2026 22:52:19 +0800 Subject: [PATCH] [fix] bugs --- requirements.txt | 1 + webui/web_demo.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/requirements.txt b/requirements.txt index 7f3512d..dabea77 100644 --- a/requirements.txt +++ b/requirements.txt @@ -6,6 +6,7 @@ transformers==4.57.6 speechbrain==1.1.0 onnxruntime==1.23.2 funasr==1.3.1 +addict==2.4.0 flask==3.0.3 flask-cors==6.0.1 flask-sock==0.7.0 diff --git a/webui/web_demo.py b/webui/web_demo.py index c8e8b84..3702fc0 100644 --- a/webui/web_demo.py +++ b/webui/web_demo.py @@ -69,7 +69,7 @@ def build_ids(prompt, history): def _mimi_decode(frames): codes = [f for f in frames if f and len(f) == 8] if not codes or not M['mimi']: return None - mc = torch.tensor(codes, dtype=torch.long).T.unsqueeze(0) + mc = torch.tensor(codes, dtype=torch.long, device=M['device']).T.unsqueeze(0) mc = torch.where(mc >= 2049, torch.zeros_like(mc), mc) with torch.no_grad(): au = M['mimi'].decode(mc).audio_values.squeeze().cpu().numpy()