funasr/datasets/llm_datasets_vicuna/datasets.py
@@ -140,7 +140,7 @@ for key, data_list in outputs.items(): if isinstance(data_list[0], torch.Tensor): if data_list[0].dtype == torch.int64: if data_list[0].dtype == torch.int64 or data_list[0].dtype == torch.int32: pad_value = self.int_pad_value else:
