OPENGIT
/
Retrieval-based-Voice-Conversion-WebUI
mirror of https://github.com/RVC-Project/Retrieval-based-Voice-Conversion-WebUI.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Format code (#214) 

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot] 2023-05-02 20:22:08 +08:00 committed by GitHub
parent 71427575c4
commit 951989117b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
config.py
View File

@ -53,7 +53,7 @@ class Config:
i_device = int(self.device.split(":")[-1]) i_device = int(self.device.split(":")[-1])
self.gpu_name = torch.cuda.get_device_name(i_device) self.gpu_name = torch.cuda.get_device_name(i_device)
if ( if (
("16" in gpu_name and "V100"not in gpu_name.upper()) ("16" in gpu_name and "V100" not in gpu_name.upper())
or "P40" in self.gpu_name.upper() or "P40" in self.gpu_name.upper()
or "1070" in self.gpu_name or "1070" in self.gpu_name
or "1080" in self.gpu_name or "1080" in self.gpu_name
@ -72,12 +72,12 @@ class Config:
else: else:
self.gpu_name = None self.gpu_name = None
self.gpu_mem = int( self.gpu_mem = int(
torch.cuda.get_device_properties(i_device).total_memory torch.cuda.get_device_properties(i_device).total_memory
/ 1024 / 1024
/ 1024 / 1024
/ 1024 / 1024
+ 0.4 + 0.4
) )
if self.gpu_mem <= 4: if self.gpu_mem <= 4:
with open("trainset_preprocess_pipeline_print.py", "r") as f: with open("trainset_preprocess_pipeline_print.py", "r") as f:
strr = f.read().replace("3.7", "3.0") strr = f.read().replace("3.7", "3.0")
@ -90,7 +90,7 @@ class Config:
print("没有发现支持的N卡, 使用CPU进行推理") print("没有发现支持的N卡, 使用CPU进行推理")
self.device = "cpu" self.device = "cpu"
self.is_half = True self.is_half = True
if self.n_cpu == 0: if self.n_cpu == 0:
self.n_cpu = cpu_count() self.n_cpu = cpu_count()