From a2f7dd634a926b59cc55c70f31b44747a7fbbe9d Mon Sep 17 00:00:00 2001 From: vipcxj Date: Tue, 12 May 2026 11:10:40 +0800 Subject: [PATCH] fix tensor not in same device error --- webui/web_demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/webui/web_demo.py b/webui/web_demo.py index c8e8b84..ec9682b 100644 --- a/webui/web_demo.py +++ b/webui/web_demo.py @@ -72,7 +72,7 @@ def _mimi_decode(frames): mc = torch.tensor(codes, dtype=torch.long).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() + au = M['mimi'].decode(mc.to(M['device'])).audio_values.squeeze().cpu().numpy() return au, mc.shape[-1] def pcm_bytes(frames, ov):