Merge pull request #8 from vipcxj/master

fix tensor not in same device error
2026-05-11 22:49:48 +08:00 · 2026-05-11 22:49:48 +08:00 · af06c632ee
parent 0d1ae504f2 c2e62c4b75
commit af06c632ee
1 changed files with 1 additions and 1 deletions
--- a/webui/web_demo.py
+++ b/webui/web_demo.py
@ -486,7 +486,7 @@ def init_model(args):
        au = torch.full((1, 8, 3), 2049, dtype=torch.long, device=args.device)
        M['model'].forward(torch.cat((au, ids.unsqueeze(1)), dim=1))
        if M['model'].audio_encoder: M['model'].audio_encoder(torch.zeros(1, 100, 560, device=args.device), torch.tensor([100], device=args.device))
-        if M['mimi']: M['mimi'].decode(torch.zeros(1, 8, 1, dtype=torch.long))
+        if M['mimi']: M['mimi'].decode(torch.zeros(1, 8, 1, dtype=torch.long, device=args.device))
    print('Warmup done!')