| funasr/datasets/openai_datasets/datasets.py | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
funasr/datasets/openai_datasets/datasets.py
@@ -162,7 +162,7 @@ if badcase_flag: continue input_ids = torch.tensor(input_ids, dtype=torch.int64) # [: self.max_token_length] attention_mask = torch.tensor([len(input_ids)], dtype=torch.int32) attention_mask = torch.tensor([1] * len(input_ids), dtype=torch.int32) labels = torch.tensor(labels, dtype=torch.int64) # [: self.max_token_length] fbank = speech[0, :, :]