ShiromiyaGamer commited on
Commit
934fe4e
1 Parent(s): b66714e

Upload dump_w2v2_feature.py

Browse files
Files changed (1) hide show
  1. dump_w2v2_feature.py +2 -1
dump_w2v2_feature.py CHANGED
@@ -65,7 +65,8 @@ class Wav2Vec2FeatureReader(object):
65
  padding_mask=None,
66
  mask=False,
67
  )
68
- feat_chunk = res["x"]
 
69
  feat.append(feat_chunk)
70
  return torch.cat(feat, 1).squeeze(0)
71
 
 
65
  padding_mask=None,
66
  mask=False,
67
  )
68
+ # Supondo que o primeiro elemento da tupla seja o que você precisa
69
+ feat_chunk = res[0] # Ajuste o índice conforme necessário
70
  feat.append(feat_chunk)
71
  return torch.cat(feat, 1).squeeze(0)
72