Spaces:
Running
Running
Update src/rvc.py
Browse files- src/rvc.py +2 -9
src/rvc.py
CHANGED
@@ -19,8 +19,8 @@ BASE_DIR = Path(__file__).resolve().parent.parent
|
|
19 |
|
20 |
class Config:
|
21 |
def __init__(self, device, is_half):
|
22 |
-
self.device =
|
23 |
-
self.is_half =
|
24 |
self.n_cpu = 0
|
25 |
self.gpu_name = None
|
26 |
self.gpu_mem = None
|
@@ -62,13 +62,6 @@ class Config:
|
|
62 |
strr = f.read().replace("3.7", "3.0")
|
63 |
with open(BASE_DIR / "src" / "trainset_preprocess_pipeline_print.py", "w") as f:
|
64 |
f.write(strr)
|
65 |
-
elif torch.backends.mps.is_available():
|
66 |
-
print("No supported N-card found, use MPS for inference")
|
67 |
-
self.device = "mps"
|
68 |
-
else:
|
69 |
-
print("No supported N-card found, use CPU for inference")
|
70 |
-
self.device = "cpu"
|
71 |
-
self.is_half = True
|
72 |
|
73 |
if self.n_cpu == 0:
|
74 |
self.n_cpu = cpu_count()
|
|
|
19 |
|
20 |
class Config:
|
21 |
def __init__(self, device, is_half):
|
22 |
+
self.device = 'cpu'
|
23 |
+
self.is_half = True
|
24 |
self.n_cpu = 0
|
25 |
self.gpu_name = None
|
26 |
self.gpu_mem = None
|
|
|
62 |
strr = f.read().replace("3.7", "3.0")
|
63 |
with open(BASE_DIR / "src" / "trainset_preprocess_pipeline_print.py", "w") as f:
|
64 |
f.write(strr)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
65 |
|
66 |
if self.n_cpu == 0:
|
67 |
self.n_cpu = cpu_count()
|