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/8/head
vipcxj 2026-05-11 22:17:12 +08:00
parent 0d1ae504f2
commit c2e62c4b75
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

@ -486,7 +486,7 @@ def init_model(args):
au = torch.full((1, 8, 3), 2049, dtype=torch.long, device=args.device) au = torch.full((1, 8, 3), 2049, dtype=torch.long, device=args.device)
M['model'].forward(torch.cat((au, ids.unsqueeze(1)), dim=1)) 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['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!') print('Warmup done!')