语帆 2 лет назад
Родитель
Сommit
6d4a5c1931
1 измененных файлов с 3 добавлено и 0 удалено
  1. 3 0
      funasr/models/lcbnet/model.py

+ 3 - 0
funasr/models/lcbnet/model.py

@@ -117,6 +117,9 @@ class LCBNet(nn.Module):
         self.specaug = specaug
         self.specaug = specaug
         self.normalize = normalize
         self.normalize = normalize
         self.encoder = encoder
         self.encoder = encoder
+        self.text_encoder = text_encoder
+        self.fusion_encoder = fusion_encoder
+        self.bias_predictor = bias_predictor
 
 
         if not hasattr(self.encoder, "interctc_use_conditioning"):
         if not hasattr(self.encoder, "interctc_use_conditioning"):
             self.encoder.interctc_use_conditioning = False
             self.encoder.interctc_use_conditioning = False