r3gm commited on
Commit
3b9943c
1 Parent(s): e0ee1f9

Update src/mdx.py

Browse files
Files changed (1) hide show
  1. 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 = 16
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