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:
@@ -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,
|
||||||
|
|||||||
Reference in New Issue
Block a user