update readme
Browse files
README.md
CHANGED
|
@@ -137,11 +137,10 @@ with torch.no_grad():
|
|
| 137 |
)
|
| 138 |
|
| 139 |
for message in processor.decode(outputs):
|
| 140 |
-
|
| 141 |
-
|
| 142 |
-
|
| 143 |
-
|
| 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 |
|