github
/
minimind-o
mirror of https://github.com/jingyaogong/minimind-o.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix tensor not in same device error

pull/9/head
vipcxj 2026-05-12 11:10:40 +08:00
parent c2e62c4b75
commit a2f7dd634a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
webui/web_demo.py
View File

@ -72,7 +72,7 @@ def _mimi_decode(frames):
mc = torch.tensor(codes, dtype=torch.long).T.unsqueeze(0) mc = torch.tensor(codes, dtype=torch.long).T.unsqueeze(0)
mc = torch.where(mc >= 2049, torch.zeros_like(mc), mc) mc = torch.where(mc >= 2049, torch.zeros_like(mc), mc)
with torch.no_grad(): 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] return au, mc.shape[-1]
def pcm_bytes(frames, ov): def pcm_bytes(frames, ov):