1
0
mirror of https://github.com/fumiama/Retrieval-based-Voice-Conversion-WebUI.git synced 2026-06-09 04:29:50 +08:00

fix v2 onnx export

This commit is contained in:
RVC-Boss
2023-06-15 15:27:51 +00:00
committed by GitHub
parent 75d7c03d41
commit 147d3c83b7

View File

@@ -550,6 +550,7 @@ class SynthesizerTrnMsNSFsidM(nn.Module):
spk_embed_dim, spk_embed_dim,
gin_channels, gin_channels,
sr, sr,
version,
**kwargs **kwargs
): ):
super().__init__() super().__init__()
@@ -573,7 +574,7 @@ class SynthesizerTrnMsNSFsidM(nn.Module):
self.gin_channels = gin_channels self.gin_channels = gin_channels
# self.hop_length = hop_length# # self.hop_length = hop_length#
self.spk_embed_dim = spk_embed_dim self.spk_embed_dim = spk_embed_dim
if self.gin_channels == 256: if version == "v1":
self.enc_p = TextEncoder256( self.enc_p = TextEncoder256(
inter_channels, inter_channels,
hidden_channels, hidden_channels,