Spaces:
Running
on
Zero
Running
on
Zero
Update src/mdx.py
Browse files- src/mdx.py +1 -1
src/mdx.py
CHANGED
@@ -258,7 +258,7 @@ def run_mdx(model_params, output_dir, model_path, filename, exclude_main=False,
|
|
258 |
device = torch.device("cpu")
|
259 |
m_threads = 2
|
260 |
if torch.cuda.is_available():
|
261 |
-
m_threads =
|
262 |
print(f"threads: {m_threads}")
|
263 |
processor_num = -1
|
264 |
|
|
|
258 |
device = torch.device("cpu")
|
259 |
m_threads = 2
|
260 |
if torch.cuda.is_available():
|
261 |
+
m_threads = 8
|
262 |
print(f"threads: {m_threads}")
|
263 |
processor_num = -1
|
264 |
|