YWMditto commited on
Commit
5008398
·
1 Parent(s): eea4939

update readme

Browse files
Files changed (1) hide show
  1. README.md +4 -5
README.md CHANGED
@@ -137,11 +137,10 @@ with torch.no_grad():
137
  )
138
 
139
  for message in processor.decode(outputs):
140
- for seg_idx, audio in enumerate(message.audio_codes_list):
141
- # audio is a waveform tensor after decode_audio_codes
142
- out_path = save_dir / f"sample{sample_idx}_seg{seg_idx}.wav"
143
- sample_idx += 1
144
- torchaudio.save(out_path, audio.unsqueeze(0), processor.model_config.sampling_rate)
145
 
146
  ```
147
 
 
137
  )
138
 
139
  for message in processor.decode(outputs):
140
+ audio = message.audio_codes_list[0]
141
+ out_path = save_dir / f"sample{sample_idx}.wav"
142
+ sample_idx += 1
143
+ torchaudio.save(out_path, audio.unsqueeze(0), processor.model_config.sampling_rate)
 
144
 
145
  ```
146