Tomtom84 commited on
Commit
9e2fbd8
verified
1 Parent(s): d44e840

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -82,7 +82,8 @@ def decode_block(b):
82
  l2.append(b[4]-16384)
83
  l3 += [b[5]-20480, b[6]-24576]
84
  codes=[torch.tensor(x,device=device).unsqueeze(0) for x in (l1,l2,l3)]
85
- audio=snac.decode(codes).squeeze().cpu().numpy()
 
86
  return (audio*32767).astype("int16").tobytes()
87
 
88
  # 5聽路聽WebSocket鈥慐ndpoint ---------------------------------------------
 
82
  l2.append(b[4]-16384)
83
  l3 += [b[5]-20480, b[6]-24576]
84
  codes=[torch.tensor(x,device=device).unsqueeze(0) for x in (l1,l2,l3)]
85
+ with torch.no_grad():
86
+ audio = snac.decode(codes).squeeze().detach().cpu().numpy()
87
  return (audio*32767).astype("int16").tobytes()
88
 
89
  # 5聽路聽WebSocket鈥慐ndpoint ---------------------------------------------