r3gm commited on
Commit
8d7b3d6
·
verified ·
1 Parent(s): 295f1b3

Update src/main.py

Browse files
Files changed (1) hide show
  1. src/main.py +1 -1
src/main.py CHANGED
@@ -247,7 +247,7 @@ def get_audio_file(song_input, is_webui, input_type, progress):
247
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
248
  compute_half = True if torch.cuda.is_available() else False
249
  config = Config(device, compute_half)
250
- hubert_model = load_hubert("cuda", config.is_half, os.path.join(rvc_models_dir, 'hubert_base.pt'))
251
  print(device, "half>>", config.is_half)
252
 
253
  # @spaces.GPU(enable_queue=True)
 
247
  device = "cuda:0" if torch.cuda.is_available() else "cpu"
248
  compute_half = True if torch.cuda.is_available() else False
249
  config = Config(device, compute_half)
250
+ hubert_model = load_hubert("cuda", config.is_half, None)
251
  print(device, "half>>", config.is_half)
252
 
253
  # @spaces.GPU(enable_queue=True)