HKUST-Audio
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -48,7 +48,7 @@ wav_tensor = torch.from_numpy(wav).float().unsqueeze(0)
|
|
48 |
with torch.no_grad():
|
49 |
# only 16khz speech
|
50 |
vq_code = model.encode_code(input_waveform=wav_tensor)
|
51 |
-
print("Code:", vq_code )
|
52 |
|
53 |
recon_wav = model.decode_code(vq_code).cpu()
|
54 |
|
@@ -57,4 +57,4 @@ sf.write("reconstructed.wav", recon_wav[0, 0, :].numpy(), sr)
|
|
57 |
print("Done! Check reconstructed.wav")
|
58 |
```
|
59 |
|
60 |
-
**If you want to train your own xcodec2 or
|
|
|
48 |
with torch.no_grad():
|
49 |
# only 16khz speech
|
50 |
vq_code = model.encode_code(input_waveform=wav_tensor)
|
51 |
+
print("Code:", vq_code )
|
52 |
|
53 |
recon_wav = model.decode_code(vq_code).cpu()
|
54 |
|
|
|
57 |
print("Done! Check reconstructed.wav")
|
58 |
```
|
59 |
|
60 |
+
**If you want to train your own xcodec2, batch inference, or large-scale code extraction, the code is released [here](https://github.com/zhenye234/X-Codec-2.0).**
|