Kangarroar commited on
Commit
bf3bc35
1 Parent(s): 8371e66

Update infer_tools/infer_tool.py

Browse files
Files changed (1) hide show
  1. infer_tools/infer_tool.py +1 -0
infer_tools/infer_tool.py CHANGED
@@ -21,6 +21,7 @@ from preprocessing.hubertinfer import Hubertencoder
21
  from utils.hparams import hparams, set_hparams
22
  from utils.pitch_utils import denorm_f0, norm_interp_f0
23
  map_location=torch.device('cpu')
 
24
  if os.path.exists("chunks_temp.json"):
25
  os.remove("chunks_temp.json")
26
 
 
21
  from utils.hparams import hparams, set_hparams
22
  from utils.pitch_utils import denorm_f0, norm_interp_f0
23
  map_location=torch.device('cpu')
24
+ os.environ["CUDA_VISIBLE_DEVICES"] = ""
25
  if os.path.exists("chunks_temp.json"):
26
  os.remove("chunks_temp.json")
27