r3gm commited on
Commit
e407867
1 Parent(s): baccc99

Update src/main.py

Browse files
Files changed (1) hide show
  1. src/main.py +2 -2
src/main.py CHANGED
@@ -192,8 +192,8 @@ 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 = 'cuda:0'
196
- config = Config(device, True)
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, config.is_half, config, rvc_model_path)
199
 
 
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, config.is_half, config, rvc_model_path)
199