yubo0306 commited on
Commit
bbd229d
·
verified ·
1 Parent(s): e60e106

Update configuration_avhubert.py

Browse files
Files changed (1) hide show
  1. configuration_avhubert.py +3 -0
configuration_avhubert.py CHANGED
@@ -51,8 +51,10 @@ class AVHubertConfig(PretrainedConfig):
51
  ctc_zero_infinity: bool = False,
52
  ctc_loss_weight: float = 0.3,
53
  special_ids: list[int] | None = None,
 
54
  **kwargs,
55
  ):
 
56
  super().__init__(**kwargs)
57
  self.label_rate = label_rate
58
  self.encoder_layers = encoder_layers
@@ -99,6 +101,7 @@ class AVHubertConfig(PretrainedConfig):
99
  self.ctc_zero_infinity = ctc_zero_infinity
100
  self.ctc_loss_weight = ctc_loss_weight
101
  self.special_ids = special_ids
 
102
 
103
  @property
104
  def encoder_config(self) -> HubertConfig:
 
51
  ctc_zero_infinity: bool = False,
52
  ctc_loss_weight: float = 0.3,
53
  special_ids: list[int] | None = None,
54
+ use_cache: bool = True,
55
  **kwargs,
56
  ):
57
+ kwargs.setdefault("is_encoder_decoder", True)
58
  super().__init__(**kwargs)
59
  self.label_rate = label_rate
60
  self.encoder_layers = encoder_layers
 
101
  self.ctc_zero_infinity = ctc_zero_infinity
102
  self.ctc_loss_weight = ctc_loss_weight
103
  self.special_ids = special_ids
104
+ self.use_cache = use_cache
105
 
106
  @property
107
  def encoder_config(self) -> HubertConfig: