poemsforaphrodite commited on
Commit
4c58375
1 Parent(s): 95831e8

Upload openvoice_app.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. openvoice_app.py +1 -0
openvoice_app.py CHANGED
@@ -14,6 +14,7 @@ args = parser.parse_args()
14
  en_ckpt_base = 'base_speakers/EN'
15
  zh_ckpt_base = 'base_speakers/ZH'
16
  ckpt_converter = 'converter'
 
17
  output_dir = 'outputs'
18
  os.makedirs(output_dir, exist_ok=True)
19
 
 
14
  en_ckpt_base = 'base_speakers/EN'
15
  zh_ckpt_base = 'base_speakers/ZH'
16
  ckpt_converter = 'converter'
17
+ device = 'cuda' if torch.cuda.is_available() else 'cpu'
18
  output_dir = 'outputs'
19
  os.makedirs(output_dir, exist_ok=True)
20