patrickvonplaten commited on
Commit
2644cfe
1 Parent(s): a8304c5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -62,7 +62,7 @@ wav = model.generate_with_chroma(descriptions, melody[None].expand(3, -1, -1), s
62
 
63
  for idx, one_wav in enumerate(wav):
64
  # Will save under {idx}.wav, with loudness normalization at -14 db LUFS.
65
- audio_write(f'{idx}', one_wav, model.sample_rate, strategy="loudness")
66
  ```
67
 
68
  ## Model details
 
62
 
63
  for idx, one_wav in enumerate(wav):
64
  # Will save under {idx}.wav, with loudness normalization at -14 db LUFS.
65
+ audio_write(f'{idx}', one_wav.cpu(), model.sample_rate, strategy="loudness")
66
  ```
67
 
68
  ## Model details