From 7cee3ab7688f079a0383aaefb7646d8db0e2ae93 Mon Sep 17 00:00:00 2001 From: Ethanfel Date: Tue, 28 Apr 2026 15:49:51 +0200 Subject: [PATCH] fix: default embedding model to EAT_LARGE Co-Authored-By: Claude Opus 4.6 --- core/audio_scan.py | 2 +- main.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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)