Eempostor
/

Harmonify

Model card Files Files and versions Community

Eempostor commited on Apr 18, 2024

Commit

b137964

·

verified ·

1 Parent(s): e195c26

Update lib/infer.py

Files changed (1) hide show

lib/infer.py +1 -1

lib/infer.py CHANGED Viewed

@@ -127,7 +127,7 @@ def infer_audio(
     ):
     os.environ["rmvpe_model_path"] = rmvpe_model_path
     os.environ["fcpe_model_path"] = fcpe_model_path
-    configs = Configs('cuda:0', False)
     vc = VC(configs)
     pth_path, index_path = get_model(model_name)
     vc_data = vc.get_vc(pth_path, protect, 0.5)

     ):
     os.environ["rmvpe_model_path"] = rmvpe_model_path
     os.environ["fcpe_model_path"] = fcpe_model_path
+    configs = Configs('cuda:0', True)
     vc = VC(configs)
     pth_path, index_path = get_model(model_name)
     vc_data = vc.get_vc(pth_path, protect, 0.5)