🐛 fix AUTO backend support

2026-01-30 20:24:31 +00:00 · 2020-12-08 12:48:15 +01:00
parent 1886dde38b
commit 232bf0d3b6
3 changed files with 7 additions and 2 deletions
--- a/spleeter/separator.py
+++ b/spleeter/separator.py
@@ -333,6 +333,11 @@ class Separator(object):
                    (Optional) string describing the waveform (e.g. filename).
        """
        backend: str = self._params['stft_backend']
+        if backend == STFTBackend.AUTO:
+            if len(tf.config.list_physical_devices('GPU')):
+                backend = STFTBackend.TENSORFLOW
+            else:
+                backend = STFTBackend.LIBROSA
        if backend == STFTBackend.TENSORFLOW:
            return self._separate_tensorflow(waveform, audio_descriptor)
        elif backend == STFTBackend.LIBROSA: