ZiyueJiang commited on
Commit
15868b7
·
1 Parent(s): 5877167

update pytorch version for ZeroGPU

Browse files
Files changed (1) hide show
  1. tts/gradio_api.py +3 -2
tts/gradio_api.py CHANGED
@@ -63,6 +63,7 @@ if __name__ == '__main__':
63
  os.system('huggingface-cli download ByteDance/MegaTTS3 --local-dir ./checkpoints --repo-type model')
64
 
65
  mp.set_start_method('spawn', force=True)
 
66
  devices = os.environ.get('CUDA_VISIBLE_DEVICES', '')
67
  if devices != '':
68
  devices = os.environ.get('CUDA_VISIBLE_DEVICES', '').split(",")
@@ -72,8 +73,8 @@ if __name__ == '__main__':
72
  devices = None
73
 
74
  num_workers = 1
75
- input_queue = mp.Queue()
76
- output_queue = mp.Queue()
77
  processes = []
78
 
79
  print("Start open workers")
 
63
  os.system('huggingface-cli download ByteDance/MegaTTS3 --local-dir ./checkpoints --repo-type model')
64
 
65
  mp.set_start_method('spawn', force=True)
66
+ mp_manager = mp.Manager()
67
  devices = os.environ.get('CUDA_VISIBLE_DEVICES', '')
68
  if devices != '':
69
  devices = os.environ.get('CUDA_VISIBLE_DEVICES', '').split(",")
 
73
  devices = None
74
 
75
  num_workers = 1
76
+ input_queue = mp_manager.Queue()
77
+ output_queue = mp_manager.Queue()
78
  processes = []
79
 
80
  print("Start open workers")