Spaces:
Configuration error
Configuration error
Kangarroar
commited on
Commit
•
bf3bc35
1
Parent(s):
8371e66
Update infer_tools/infer_tool.py
Browse files
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 |
|