From aea223d41e6bbc01aaaacdf477a36059c1697513 Mon Sep 17 00:00:00 2001
From: "github-actions[bot]"
 <41898282+github-actions[bot]@users.noreply.github.com>
Date: Sun, 14 Apr 2024 15:53:07 +0900
Subject: [PATCH] chore(format): run black on dev (#1964)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
---
 infer/lib/infer_pack/models_onnx.py | 18 +++++++++---------
 tools/export_onnx.py                |  2 +-
 2 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/infer/lib/infer_pack/models_onnx.py b/infer/lib/infer_pack/models_onnx.py
index f1b846a..54ee652 100644
--- a/infer/lib/infer_pack/models_onnx.py
+++ b/infer/lib/infer_pack/models_onnx.py
@@ -636,15 +636,15 @@ class SynthesizerTrnMsNSFsidM(nn.Module):
         # self.hop_length = hop_length#
         self.spk_embed_dim = spk_embed_dim
         self.enc_p = TextEncoder(
-                encoder_dim,
-                inter_channels,
-                hidden_channels,
-                filter_channels,
-                n_heads,
-                n_layers,
-                kernel_size,
-                float(p_dropout),
-            )
+            encoder_dim,
+            inter_channels,
+            hidden_channels,
+            filter_channels,
+            n_heads,
+            n_layers,
+            kernel_size,
+            float(p_dropout),
+        )
         self.dec = GeneratorNSF(
             inter_channels,
             resblock,
diff --git a/tools/export_onnx.py b/tools/export_onnx.py
index 9380aa7..099948c 100644
--- a/tools/export_onnx.py
+++ b/tools/export_onnx.py
@@ -21,7 +21,7 @@ if __name__ == "__main__":
     device = "cpu"  # 导出时设备（不影响使用模型）
 
     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）
     net_g.load_state_dict(cpt["weight"], strict=False)
     input_names = ["phone", "phone_lengths", "pitch", "pitchf", "ds", "rnd"]