Format code (#989)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

Author: github-actions[bot]

MDXNet.py

@@ -83,12 +83,13 @@ def get_models(device, dim_f, dim_t, n_fft):
 
 warnings.filterwarnings("ignore")
 import sys
+
 now_dir = os.getcwd()
 sys.path.append(now_dir)
 from config import Config
 
 cpu = torch.device("cpu")
-device=Config().device
+device = Config().device
 # if torch.cuda.is_available():
 #     device = torch.device("cuda:0")
 # elif torch.backends.mps.is_available():
@@ -104,10 +105,15 @@ def __init__(self, args):
             device=cpu, dim_f=args.dim_f, dim_t=args.dim_t, n_fft=args.n_fft
         )
         import onnxruntime as ort
+
         print(ort.get_available_providers())
         self.model = ort.InferenceSession(
             os.path.join(args.onnx, self.model_.target_name + ".onnx"),
-            providers=["CUDAExecutionProvider", "DmlExecutionProvider","CPUExecutionProvider"],
+            providers=[
+                "CUDAExecutionProvider",
+                "DmlExecutionProvider",
+                "CPUExecutionProvider",
+            ],
         )
         print("onnx load done")
 

config.py

@@ -36,7 +36,7 @@ def __init__(self):
             self.iscolab,
             self.noparallel,
             self.noautoopen,
-            self.dml
+            self.dml,
         ) = self.arg_parse()
         self.instead = ""
         self.x_pad, self.x_query, self.x_center, self.x_max = self.device_config()
@@ -71,7 +71,7 @@ def arg_parse() -> tuple:
             cmd_opts.colab,
             cmd_opts.noparallel,
             cmd_opts.noautoopen,
-            cmd_opts.dml
+            cmd_opts.dml,
         )
 
     # has_mps is only available in nightly pytorch (for now) and MasOS 12.3+.
@@ -149,26 +149,38 @@ def device_config(self) -> tuple:
         if self.dml:
             print("use DirectML instead")
             try:
-                os.rename("runtime\Lib\site-packages\onnxruntime","runtime\Lib\site-packages\onnxruntime-cuda")
+                os.rename(
+                    "runtime\Lib\site-packages\onnxruntime",
+                    "runtime\Lib\site-packages\onnxruntime-cuda",
+                )
             except:
                 pass
             try:
-                os.rename("runtime\Lib\site-packages\onnxruntime-dml","runtime\Lib\site-packages\onnxruntime")
+                os.rename(
+                    "runtime\Lib\site-packages\onnxruntime-dml",
+                    "runtime\Lib\site-packages\onnxruntime",
+                )
             except:
-
                 pass
             import torch_directml
+
             self.device = torch_directml.device(torch_directml.default_device())
             self.is_half = False
         else:
             if self.instead:
                 print(f"use {self.instead} instead")
             try:
-                os.rename("runtime\Lib\site-packages\onnxruntime","runtime\Lib\site-packages\onnxruntime-cuda")
+                os.rename(
+                    "runtime\Lib\site-packages\onnxruntime",
+                    "runtime\Lib\site-packages\onnxruntime-cuda",
+                )
             except:
                 pass
             try:
-                os.rename("runtime\Lib\site-packages\onnxruntime-dml","runtime\Lib\site-packages\onnxruntime")
+                os.rename(
+                    "runtime\Lib\site-packages\onnxruntime-dml",
+                    "runtime\Lib\site-packages\onnxruntime",
+                )
             except:
                 pass
         return x_pad, x_query, x_center, x_max

extract_f0_rmvpe_dml.py

@@ -10,6 +10,7 @@
 
 exp_dir = sys.argv[1]
 import torch_directml
+
 device = torch_directml.device(torch_directml.default_device())
 f = open("%s/extract_f0_feature.log" % exp_dir, "a+")
 

extract_feature_print.py

@@ -3,7 +3,7 @@
 os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
 os.environ["PYTORCH_MPS_HIGH_WATERMARK_RATIO"] = "0.0"
 
-device=sys.argv[1]
+device = sys.argv[1]
 n_part = int(sys.argv[2])
 i_part = int(sys.argv[3])
 if len(sys.argv) == 6:
@@ -20,20 +20,23 @@
 import numpy as np
 import fairseq
 
-if("privateuseone"not in device):
+if "privateuseone" not in device:
     device = "cpu"
     if torch.cuda.is_available():
         device = "cuda"
     elif torch.backends.mps.is_available():
         device = "mps"
 else:
     import torch_directml
+
     device = torch_directml.device(torch_directml.default_device())
+
     def forward_dml(ctx, x, scale):
         ctx.scale = scale
         res = x.clone().detach()
         return res
-    fairseq.modules.grad_multiply.GradMultiply.forward=forward_dml
+
+    fairseq.modules.grad_multiply.GradMultiply.forward = forward_dml
 
 f = open("%s/extract_f0_feature.log" % exp_dir, "a+")
 

gui_v1.py

@@ -1,5 +1,6 @@
-import os, sys,pdb
-os.environ["OMP_NUM_THREADS"]="2"
+import os, sys, pdb
+
+os.environ["OMP_NUM_THREADS"] = "2"
 if sys.platform == "darwin":
     os.environ["PYTORCH_ENABLE_MPS_FALLBACK"] = "1"
 
@@ -47,8 +48,9 @@ def run(self):
     import torchaudio.transforms as tat
     from i18n import I18nAuto
     import rvc_for_realtime
+
     i18n = I18nAuto()
-    device=rvc_for_realtime.config.device
+    device = rvc_for_realtime.config.device
     # device = torch.device(
     #     "cuda"
     #     if torch.cuda.is_available()
@@ -61,7 +63,6 @@ def run(self):
     for _ in range(n_cpu):
         Harvest(inp_q, opt_q).start()
 
-
     class GUIConfig:
         def __init__(self) -> None:
             self.pth_path: str = ""