Politrees commited on
Commit
278bd08
β€’
1 Parent(s): 21b151b

Update src/rvc.py

Browse files
Files changed (1) hide show
  1. 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 = device
23
- self.is_half = 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()