From 43d109e9c47a2024eac2e2efdbef2f6897827bf6 Mon Sep 17 00:00:00 2001 From: Tps-F Date: Sat, 19 Aug 2023 10:17:30 +0000 Subject: [PATCH] Apply Code Formatter Change --- config.py | 35 +++++++++++++++++++++++++++++------ infer-web.py | 5 ++++- 2 files changed, 33 insertions(+), 7 deletions(-) diff --git a/config.py b/config.py index 0261b02..53f8135 100644 --- a/config.py +++ b/config.py @@ -148,28 +148,51 @@ class Config: x_max = 32 if self.dml: print("use DirectML instead") - if(os.path.exists("runtime\Lib\site-packages\onnxruntime\capi\DirectML.dll")==False): + if ( + os.path.exists( + "runtime\Lib\site-packages\onnxruntime\capi\DirectML.dll" + ) + == False + ): 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") - if(os.path.exists("runtime\Lib\site-packages\onnxruntime\capi\onnxruntime_providers_cuda.dll")==False): + if ( + os.path.exists( + "runtime\Lib\site-packages\onnxruntime\capi\onnxruntime_providers_cuda.dll" + ) + == False + ): try: - os.rename("runtime\Lib\site-packages\onnxruntime", "runtime\Lib\site-packages\onnxruntime-dml") + os.rename( + "runtime\Lib\site-packages\onnxruntime", + "runtime\Lib\site-packages\onnxruntime-dml", + ) except: pass try: - os.rename("runtime\Lib\site-packages\onnxruntime-cuda", "runtime\Lib\site-packages\onnxruntime") + os.rename( + "runtime\Lib\site-packages\onnxruntime-cuda", + "runtime\Lib\site-packages\onnxruntime", + ) except: pass return x_pad, x_query, x_center, x_max diff --git a/infer-web.py b/infer-web.py index 742299b..c5fd117 100644 --- a/infer-web.py +++ b/infer-web.py @@ -239,7 +239,10 @@ def vc_single( times[0], times[1], times[2], - ), (resample_sr if resample_sr >= 16000 and tgt_sr != resample_sr else tgt_sr, audio_opt) + ), ( + resample_sr if resample_sr >= 16000 and tgt_sr != resample_sr else tgt_sr, + audio_opt, + ) except: info = traceback.format_exc() print(info)