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 |
|