This commit is contained in:
维石 2024-09-30 14:51:59 +08:00
parent a62ef27d71
commit 1ca3c42282

View File

@ -1082,7 +1082,7 @@ class SenseVoiceQuantizedEncoderPitch(nn.Module):
self.normalized_quant_input = normalized_quant_input
self.quantizer = self.build_quantizer(quantizer_config)
self.pitch_predictor = torch.Linear(units, 1)
self.pitch_predictor = torch.nn.Linear(units, 1)
self.pitch_act = torch.nn.ReLU()
def build_quantizer(self, vq_config):