KingNish commited on
Commit
185e94b
·
verified ·
1 Parent(s): fd82b48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -290,7 +290,7 @@ def generate_music(
290
  for npy in stage1_output_set:
291
  codec_result = np.load(npy)
292
  decodec_rlt = []
293
- with torch.grad():
294
  decoded_waveform = codec_model.decode(
295
  torch.as_tensor(codec_result.astype(np.int16), dtype=torch.long).unsqueeze(0).permute(1, 0, 2).to(
296
  device))
 
290
  for npy in stage1_output_set:
291
  codec_result = np.load(npy)
292
  decodec_rlt = []
293
+ with torch.no_grad():
294
  decoded_waveform = codec_model.decode(
295
  torch.as_tensor(codec_result.astype(np.int16), dtype=torch.long).unsqueeze(0).permute(1, 0, 2).to(
296
  device))