ffmpeg

游雁

2024-03-18 322f3d5ab9bf7c64359d3a5e50dac1d9f7de4534

 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: