diff --git a/core/audio_scan.py b/core/audio_scan.py index 7a31712..92121c1 100644 --- a/core/audio_scan.py +++ b/core/audio_scan.py @@ -67,7 +67,7 @@ _EMBED_MODELS = { "EAT": 768, "EAT_LARGE": 1024, } -_DEFAULT_EMBED_MODEL = "WAV2VEC2_BASE" +_DEFAULT_EMBED_MODEL = "EAT_LARGE" _BEATS_CHECKPOINT = os.path.join( _DL_CACHE_DIR, "huggingface", "hub", diff --git a/main.py b/main.py index e557e94..4bc7f52 100755 --- a/main.py +++ b/main.py @@ -465,7 +465,7 @@ class TrainDialog(QDialog): self._cmb_model = QComboBox() for name in _EMBED_MODELS: self._cmb_model.addItem(name) - self._cmb_model.setCurrentText("HUBERT_XLARGE") + self._cmb_model.setCurrentText("EAT_LARGE") form.addRow("Model:", self._cmb_model) # Auto-negative margin (0 = disabled)