yuzu/spleeter
1
0
Fork 0
mirror of https://github.com/YuzuZensai/spleeter.git synced 2026-01-31 14:58:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

🐛 fix model download

Browse Source
This commit is contained in:
Faylixe
2020-12-08 13:46:20 +01:00
parent a4d668b612
commit 2c286f255e
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
spleeter/audio/__init__.py
View File

@@ -47,5 +47,5 @@ class STFTBackend(str, Enum):
if backend == cls.AUTO: if backend == cls.AUTO:
if len(tf.config.list_physical_devices('GPU')): if len(tf.config.list_physical_devices('GPU')):
return cls.TENSORFLOW return cls.TENSORFLOW
return STFTBackend.LIBROSA return cls.LIBROSA
return backend return backend

2
spleeter/model/provider/github.py
View File

@@ -140,7 +140,7 @@ class GithubModelProvider(ModelProvider):
url = f'{url}.tar.gz' url = f'{url}.tar.gz'
logger.info(f'Downloading model archive {url}') logger.info(f'Downloading model archive {url}')
with httpx.Client(http2=True) as client: with httpx.Client(http2=True) as client:
with client.strema('GET', url) as response: with client.stream('GET', url) as response:
response.raise_for_status() response.raise_for_status()
archive = NamedTemporaryFile(delete=False) archive = NamedTemporaryFile(delete=False)
try: try: