alihan/neutts-air
1
0
Fork 0
mirror of https://github.com/neuphonic/neutts-air.git synced 2025-10-10 02:44:44 +03:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #29 from harryjulian/encoder-patch

Browse Source
This commit is contained in:
Jiameng Gao
2025-10-07 20:09:30 +01:00
committed by GitHub
parent 23d75bc9dc 36631b7787
commit 170bd67d92
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
neuttsair/neutts.py
View File

@@ -137,7 +137,8 @@ class NeuTTSAir:
def encode_reference(self, ref_audio_path: str | Path):
wav, _ = librosa.load(ref_audio_path, sr=16000, mono=True)
wav_tensor = torch.from_numpy(wav).float().unsqueeze(0).unsqueeze(0) # [1, 1, T]
ref_codes = self.codec.encode_code(audio_or_path=wav_tensor).squeeze(0).squeeze(0)
with torch.no_grad():
ref_codes = self.codec.encode_code(audio_or_path=wav_tensor).squeeze(0).squeeze(0)
return ref_codes
def _decode(self, codes: str):