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):