Update app.py
Browse files
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.
|
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))
|