Update infer_tools/infer_tool.py

infer_tools/infer_tool.py

@@ -21,7 +21,11 @@ from preprocessing.hubertinfer import Hubertencoder
 from utils.hparams import hparams, set_hparams
 from utils.pitch_utils import denorm_f0, norm_interp_f0
 map_location=torch.device('cpu')
-os.environ["CUDA_VISIBLE_DEVICES"] = ""
+# Enable JIT mode
+torch.backends.intel.set_jit_enabled(True)
+
+# Enable IPEX
+torch.backends.intel.set_use_ipex(True)
 if os.path.exists("chunks_temp.json"):
     os.remove("chunks_temp.json")