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()
|