Rename configs to it's respective models

Ahmedsaed · Aug 1, 2024 · 10cbde6 · 10cbde6
1 parent ee376df
commit 10cbde6
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/src/fairseq2/recipes/eval/__init__.py b/src/fairseq2/recipes/eval/__init__.py
@@ -6,7 +6,7 @@
 
 from fairseq2.logging import get_log_writer
 from fairseq2.recipes.cli import Cli, CliGroup, RecipeCommandHandler
-from fairseq2.recipes.eval.configs import hf_presets
+from fairseq2.recipes.eval.configs import wav2vec2_presets, whisper_presets
 
 log = get_log_writer(__name__)
 
@@ -16,7 +16,7 @@ def _add_wav2vev2_asr_eval_cli(group: CliGroup) -> None:
 
     handler = RecipeCommandHandler(
         load_wav2vec2_asr_evaluator,
-        preset_configs=hf_presets,
+        preset_configs=wav2vec2_presets,
         default_preset="librispeech_asr",
     )
     group.add_command(

diff --git a/src/fairseq2/recipes/eval/asr.py b/src/fairseq2/recipes/eval/asr.py
@@ -26,7 +26,7 @@
 from fairseq2.models.sequence import SequenceBatch
 from fairseq2.models.wav2vec2.asr import load_wav2vec2_asr_model
 from fairseq2.nn.padding import get_seqs_and_padding_mask
-from fairseq2.recipes.eval.configs import HFEvalConfig, hf_presets
+from fairseq2.recipes.eval.configs import HFEvalConfig, wav2vec2_presets, whisper_presets
 from fairseq2.recipes.evaluator import HFEvaluator
 from fairseq2.recipes.utils.setup import setup_root_gang
 from fairseq2.typing import META, DataType

diff --git a/src/fairseq2/recipes/eval/configs.py b/src/fairseq2/recipes/eval/configs.py
@@ -20,4 +20,5 @@ class HFEvalConfig:
     """The name of the model to evaluate."""
 
 
-hf_presets = ConfigRegistry[HFEvalConfig]()
+wav2vec2_presets = ConfigRegistry[HFEvalConfig]()
+whisper_presets = ConfigRegistry[HFEvalConfig]()