fix typo

wenet-e2e · Feb 7, 2025 · e644da8 · e644da8
1 parent 0625895
commit e644da8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/wenet/firered/model.py b/wenet/firered/model.py
@@ -22,7 +22,7 @@
 from wenet.utils.common import IGNORE_ID
 
 
-class FirereadModel(ASRModel):
+class FireReadModel(ASRModel):
 
     def __init__(
         self,

diff --git a/wenet/utils/init_model.py b/wenet/utils/init_model.py
@@ -24,7 +24,7 @@
 from wenet.finetune.lora.utils import (inject_lora_to_model,
                                        mark_only_lora_as_trainable)
 from wenet.firered.encoder import FireRedConformerEncoder
-from wenet.firered.model import FirereadModel
+from wenet.firered.model import FireReadModel
 from wenet.k2.model import K2Model
 from wenet.paraformer.cif import Cif
 from wenet.paraformer.layers import SanmDecoder, SanmEncoder
@@ -83,7 +83,7 @@
     "asr_model": ASRModel,
     "ctl_model": CTLModel,
     "whisper": Whisper,
-    "firered": FirereadModel,
+    "firered": FireReadModel,
     "k2_model": K2Model,
     "transducer": Transducer,
     'paraformer': Paraformer,