File tree 1 file changed +4
-1
lines changed
audio_separator/separator 1 file changed +4
-1
lines changed Original file line number Diff line number Diff line change @@ -171,6 +171,9 @@ def log_system_info(self):
171
171
python_version = platform .python_version ()
172
172
self .logger .info (f"Python Version: { python_version } " )
173
173
174
+ pytorch_version = torch .__version__
175
+ self .logger .info (f"PyTorch Version: { pytorch_version } " )
176
+
174
177
def check_ffmpeg_installed (self ):
175
178
"""
176
179
This method checks if ffmpeg is installed and logs its version.
@@ -245,7 +248,7 @@ def configure_mps(self, ort_providers):
245
248
246
249
if "CoreMLExecutionProvider" in ort_providers :
247
250
self .logger .info ("ONNXruntime has CoreMLExecutionProvider available, enabling acceleration" )
248
- self .onnx_execution_provider = ["CoreMLExecutionProvider" , "CPUExecutionProvider" ]
251
+ self .onnx_execution_provider = ["CoreMLExecutionProvider" ]
249
252
else :
250
253
self .logger .warning ("CoreMLExecutionProvider not available in ONNXruntime, so acceleration will NOT be enabled" )
251
254
You can’t perform that action at this time.
0 commit comments