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

add no_grad to reference encoding

Browse Source
This commit is contained in:
harryjulian
2025-10-07 20:00:04 +01:00
parent 23d75bc9dc
commit 36631b7787
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):