Skip to content

Commit ac1076d

Browse files
committed
Log pytorch version, remove CPU exec provider for MPS now FFT is implemented in PyTorch for MPS
1 parent ce73de1 commit ac1076d

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

audio_separator/separator/separator.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,9 @@ def log_system_info(self):
171171
python_version = platform.python_version()
172172
self.logger.info(f"Python Version: {python_version}")
173173

174+
pytorch_version = torch.__version__
175+
self.logger.info(f"PyTorch Version: {pytorch_version}")
176+
174177
def check_ffmpeg_installed(self):
175178
"""
176179
This method checks if ffmpeg is installed and logs its version.
@@ -245,7 +248,7 @@ def configure_mps(self, ort_providers):
245248

246249
if "CoreMLExecutionProvider" in ort_providers:
247250
self.logger.info("ONNXruntime has CoreMLExecutionProvider available, enabling acceleration")
248-
self.onnx_execution_provider = ["CoreMLExecutionProvider", "CPUExecutionProvider"]
251+
self.onnx_execution_provider = ["CoreMLExecutionProvider"]
249252
else:
250253
self.logger.warning("CoreMLExecutionProvider not available in ONNXruntime, so acceleration will NOT be enabled")
251254

0 commit comments

Comments
 (0)