yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-06 04:32:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Changed gpu availability function to avoid deprectation warning

Browse Source
This commit is contained in:
romi1502
2020-09-25 13:15:19 +02:00
parent af21c21b01
commit c74ab33905
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
spleeter/separator.py
View File

@@ -69,7 +69,7 @@ def get_backend(backend):
# print("USING TENSORFLOW BACKEND !!!!!!")
# return "tensorflow"
if backend == "auto":
return "tensorflow" if tf.test.is_gpu_available() else "librosa"
return "tensorflow" if len(tf.config.list_physical_devices('GPU')) else "librosa"
return backend