mirror of
https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git
synced 2025-05-06 20:01:37 +08:00
chore(format): run black on dev (#1964)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
parent
597e0a97d0
commit
aea223d41e
@ -636,15 +636,15 @@ class SynthesizerTrnMsNSFsidM(nn.Module):
|
|||||||
# self.hop_length = hop_length#
|
# self.hop_length = hop_length#
|
||||||
self.spk_embed_dim = spk_embed_dim
|
self.spk_embed_dim = spk_embed_dim
|
||||||
self.enc_p = TextEncoder(
|
self.enc_p = TextEncoder(
|
||||||
encoder_dim,
|
encoder_dim,
|
||||||
inter_channels,
|
inter_channels,
|
||||||
hidden_channels,
|
hidden_channels,
|
||||||
filter_channels,
|
filter_channels,
|
||||||
n_heads,
|
n_heads,
|
||||||
n_layers,
|
n_layers,
|
||||||
kernel_size,
|
kernel_size,
|
||||||
float(p_dropout),
|
float(p_dropout),
|
||||||
)
|
)
|
||||||
self.dec = GeneratorNSF(
|
self.dec = GeneratorNSF(
|
||||||
inter_channels,
|
inter_channels,
|
||||||
resblock,
|
resblock,
|
||||||
|
@ -21,7 +21,7 @@ if __name__ == "__main__":
|
|||||||
device = "cpu" # 导出时设备(不影响使用模型)
|
device = "cpu" # 导出时设备(不影响使用模型)
|
||||||
|
|
||||||
net_g = SynthesizerTrnMsNSFsidM(
|
net_g = SynthesizerTrnMsNSFsidM(
|
||||||
*cpt["config"], is_half=False, encoder_dim = encoder_dim
|
*cpt["config"], is_half=False, encoder_dim=encoder_dim
|
||||||
) # fp32导出(C++要支持fp16必须手动将内存重新排列所以暂时不用fp16)
|
) # fp32导出(C++要支持fp16必须手动将内存重新排列所以暂时不用fp16)
|
||||||
net_g.load_state_dict(cpt["weight"], strict=False)
|
net_g.load_state_dict(cpt["weight"], strict=False)
|
||||||
input_names = ["phone", "phone_lengths", "pitch", "pitchf", "ds", "rnd"]
|
input_names = ["phone", "phone_lengths", "pitch", "pitchf", "ds", "rnd"]
|
||||||
|
Loading…
x
Reference in New Issue
Block a user