Update FcpePitchExtractor.py

This commit is contained in:
icecoins 2024-01-12 02:28:17 +08:00 committed by GitHub
parent 1cf9be54c7
commit 8f230e5c45
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -26,7 +26,7 @@ class FcpePitchExtractor(PitchExtractor):
f0_mel_max = 1127 * np.log(1 + f0_max / 700)
f0 = self.fcpe.infer(
audio.to(self.device_fcpe).unsqueeze(0).float(),
audio.to(self.device).unsqueeze(0).float(),
sr=16000,
decoder_mode="local_argmax",
threshold=0.006,