fix: default embedding model to EAT_LARGE
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
+1
-1
@@ -67,7 +67,7 @@ _EMBED_MODELS = {
|
|||||||
"EAT": 768,
|
"EAT": 768,
|
||||||
"EAT_LARGE": 1024,
|
"EAT_LARGE": 1024,
|
||||||
}
|
}
|
||||||
_DEFAULT_EMBED_MODEL = "WAV2VEC2_BASE"
|
_DEFAULT_EMBED_MODEL = "EAT_LARGE"
|
||||||
|
|
||||||
_BEATS_CHECKPOINT = os.path.join(
|
_BEATS_CHECKPOINT = os.path.join(
|
||||||
_DL_CACHE_DIR, "huggingface", "hub",
|
_DL_CACHE_DIR, "huggingface", "hub",
|
||||||
|
|||||||
@@ -465,7 +465,7 @@ class TrainDialog(QDialog):
|
|||||||
self._cmb_model = QComboBox()
|
self._cmb_model = QComboBox()
|
||||||
for name in _EMBED_MODELS:
|
for name in _EMBED_MODELS:
|
||||||
self._cmb_model.addItem(name)
|
self._cmb_model.addItem(name)
|
||||||
self._cmb_model.setCurrentText("HUBERT_XLARGE")
|
self._cmb_model.setCurrentText("EAT_LARGE")
|
||||||
form.addRow("Model:", self._cmb_model)
|
form.addRow("Model:", self._cmb_model)
|
||||||
|
|
||||||
# Auto-negative margin (0 = disabled)
|
# Auto-negative margin (0 = disabled)
|
||||||
|
|||||||
Reference in New Issue
Block a user