fix tensor not in same device error

2026-05-12 11:10:40 +08:00 · 2026-05-12 11:10:40 +08:00 · a2f7dd634a
parent c2e62c4b75
commit a2f7dd634a
1 changed files with 1 additions and 1 deletions
--- 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):