Update lib/infer.py
Browse files- lib/infer.py +4 -4
lib/infer.py
CHANGED
|
@@ -102,7 +102,7 @@ def infer_audio(
|
|
| 102 |
model_name,
|
| 103 |
audio_path,
|
| 104 |
f0_change=0,
|
| 105 |
-
f0_method="rmvpe
|
| 106 |
min_pitch="50",
|
| 107 |
max_pitch="1100",
|
| 108 |
crepe_hop_length=128,
|
|
@@ -121,9 +121,9 @@ def infer_audio(
|
|
| 121 |
f0_autotune=False,
|
| 122 |
audio_format="wav",
|
| 123 |
resample_sr=0,
|
| 124 |
-
hubert_model_path="
|
| 125 |
-
rmvpe_model_path="
|
| 126 |
-
fcpe_model_path="
|
| 127 |
):
|
| 128 |
os.environ["rmvpe_model_path"] = rmvpe_model_path
|
| 129 |
os.environ["fcpe_model_path"] = fcpe_model_path
|
|
|
|
| 102 |
model_name,
|
| 103 |
audio_path,
|
| 104 |
f0_change=0,
|
| 105 |
+
f0_method="rmvpe",
|
| 106 |
min_pitch="50",
|
| 107 |
max_pitch="1100",
|
| 108 |
crepe_hop_length=128,
|
|
|
|
| 121 |
f0_autotune=False,
|
| 122 |
audio_format="wav",
|
| 123 |
resample_sr=0,
|
| 124 |
+
hubert_model_path="hubert_base.pt",
|
| 125 |
+
rmvpe_model_path="rmvpe.pt",
|
| 126 |
+
fcpe_model_path="fcpe.pt"
|
| 127 |
):
|
| 128 |
os.environ["rmvpe_model_path"] = rmvpe_model_path
|
| 129 |
os.environ["fcpe_model_path"] = fcpe_model_path
|