Spaces:
Sleeping
Sleeping
ibcplateformes Claude Opus 4.6 commited on
Commit ·
81bdd2b
1
Parent(s): 447db6e
Restore full quality params now that GPU is available
Browse files- Max slices: 80 (use all audio)
- Batch size: 8
- Default epochs: 20
- Slider max: 50
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
- app.py +4 -4
- pipeline/training.py +3 -3
app.py
CHANGED
|
@@ -112,7 +112,7 @@ def train_voice_model(audio_file, model_name, epochs, progress=gr.Progress()):
|
|
| 112 |
model_name=model_name,
|
| 113 |
epochs=int(epochs),
|
| 114 |
sample_rate=40000,
|
| 115 |
-
batch_size=
|
| 116 |
progress_callback=progress_callback,
|
| 117 |
)
|
| 118 |
|
|
@@ -295,10 +295,10 @@ with gr.Blocks(
|
|
| 295 |
)
|
| 296 |
train_epochs = gr.Slider(
|
| 297 |
minimum=5,
|
| 298 |
-
maximum=
|
| 299 |
-
value=
|
| 300 |
step=5,
|
| 301 |
-
label="Nombre d'époques (
|
| 302 |
)
|
| 303 |
train_btn = gr.Button(
|
| 304 |
"Lancer l'entraînement",
|
|
|
|
| 112 |
model_name=model_name,
|
| 113 |
epochs=int(epochs),
|
| 114 |
sample_rate=40000,
|
| 115 |
+
batch_size=8,
|
| 116 |
progress_callback=progress_callback,
|
| 117 |
)
|
| 118 |
|
|
|
|
| 295 |
)
|
| 296 |
train_epochs = gr.Slider(
|
| 297 |
minimum=5,
|
| 298 |
+
maximum=50,
|
| 299 |
+
value=20,
|
| 300 |
step=5,
|
| 301 |
+
label="Nombre d'époques (plus = meilleure qualité, ~3-5 min avec GPU)",
|
| 302 |
)
|
| 303 |
train_btn = gr.Button(
|
| 304 |
"Lancer l'entraînement",
|
pipeline/training.py
CHANGED
|
@@ -95,7 +95,7 @@ def preprocess(model_name: str, audio_path: str, sample_rate: int = 40000):
|
|
| 95 |
segment_len_16k = int(3.5 * 16000)
|
| 96 |
hop_16k = int(3.0 * 16000)
|
| 97 |
|
| 98 |
-
MAX_SLICES =
|
| 99 |
|
| 100 |
n_slices = 0
|
| 101 |
idx = 0
|
|
@@ -229,8 +229,8 @@ def extract_features(model_name: str, sample_rate: int = 40000, f0_method: str =
|
|
| 229 |
def train_model(
|
| 230 |
model_name: str,
|
| 231 |
sample_rate: int = 40000,
|
| 232 |
-
total_epochs: int =
|
| 233 |
-
batch_size: int =
|
| 234 |
):
|
| 235 |
"""
|
| 236 |
Train RVC v2 model. Runs IN-PROCESS with mp.Process patched to avoid
|
|
|
|
| 95 |
segment_len_16k = int(3.5 * 16000)
|
| 96 |
hop_16k = int(3.0 * 16000)
|
| 97 |
|
| 98 |
+
MAX_SLICES = 80 # Use all available audio for better quality
|
| 99 |
|
| 100 |
n_slices = 0
|
| 101 |
idx = 0
|
|
|
|
| 229 |
def train_model(
|
| 230 |
model_name: str,
|
| 231 |
sample_rate: int = 40000,
|
| 232 |
+
total_epochs: int = 20,
|
| 233 |
+
batch_size: int = 8,
|
| 234 |
):
|
| 235 |
"""
|
| 236 |
Train RVC v2 model. Runs IN-PROCESS with mp.Process patched to avoid
|