diff --git a/textless/data/hubert_feature_reader.py b/textless/data/hubert_feature_reader.py index 4c774b5..45af901 100644 --- a/textless/data/hubert_feature_reader.py +++ b/textless/data/hubert_feature_reader.py @@ -66,6 +66,8 @@ def get_features(self, x): feat = [] for start in range(0, x.size(1), self.max_chunk): x_chunk = x[:, start : start + self.max_chunk] + if x_chunk.size(1) < 10: + continue feat_chunk, _ = self.model.extract_features( source=x_chunk, padding_mask=None,