logeshr24 commited on
Commit
d7a187c
1 Parent(s): 3e80b7b

fix: cpu to cuda

Browse files
Files changed (1) hide show
  1. src/main.py +1 -1
src/main.py CHANGED
@@ -192,7 +192,7 @@ def preprocess_song(song_input, mdx_model_params, song_id, is_webui, input_type,
192
 
193
  def voice_change(voice_model, vocals_path, output_path, pitch_change, f0_method, index_rate, filter_radius, rms_mix_rate, protect, crepe_hop_length, is_webui):
194
  rvc_model_path, rvc_index_path = get_rvc_model(voice_model, is_webui)
195
- device = 'cpu'
196
  config = Config(device, False)
197
  hubert_model = load_hubert(device, config.is_half, os.path.join(rvc_models_dir, 'hubert_base.pt'))
198
  cpt, version, net_g, tgt_sr, vc = get_vc(device, False, config, rvc_model_path)
 
192
 
193
  def voice_change(voice_model, vocals_path, output_path, pitch_change, f0_method, index_rate, filter_radius, rms_mix_rate, protect, crepe_hop_length, is_webui):
194
  rvc_model_path, rvc_index_path = get_rvc_model(voice_model, is_webui)
195
+ device = 'cuda:0'
196
  config = Config(device, False)
197
  hubert_model = load_hubert(device, config.is_half, os.path.join(rvc_models_dir, 'hubert_base.pt'))
198
  cpt, version, net_g, tgt_sr, vc = get_vc(device, False, config, rvc_model_path)