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

🐛 fix resolution

Browse Source
This commit is contained in:
Faylixe
2020-12-08 13:28:10 +01:00
parent 96901a8393
commit 72e74aaf22
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
spleeter/audio/__init__.py
View File

@@ -42,7 +42,8 @@ class STFTBackend(str, Enum):
@classmethod @classmethod
def resolve(cls: type, backend: str) -> str: def resolve(cls: type, backend: str) -> str:
if backend not in cls.__members__.items(): backend = backend.upper()
if backend not in cls.__members__.keys():
raise ValueError(f'Unsupported backend {backend}') raise ValueError(f'Unsupported backend {backend}')
if backend == cls.AUTO: if backend == cls.AUTO:
if len(tf.config.list_physical_devices('GPU')): if len(tf.config.list_physical_devices('GPU')):