Spaces:
Paused
Paused
fix: cpu to cuda
Browse files- 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 = '
|
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)
|