XiaoBai1221 commited on
Commit
973652d
·
1 Parent(s): 88e2bb2

fix: update all_tied_weights_keys to return a dict to resolve keys attribute error

Browse files
models/emotion_recognition/emotion.py CHANGED
@@ -54,7 +54,7 @@ class HubertForSpeechClassification(HubertPreTrainedModel):
54
 
55
  @property
56
  def all_tied_weights_keys(self):
57
- return []
58
 
59
  def forward(self, x):
60
  outputs = self.hubert(x)
 
54
 
55
  @property
56
  def all_tied_weights_keys(self):
57
+ return {}
58
 
59
  def forward(self, x):
60
  outputs = self.hubert(x)