SenseVoice对富文本解码的参数 (#1608)

gaochangfeng

2024-04-11 fce4e1d1b48f23cd8332e60afce3df8d6209a6a7

 funasr/datasets/audio_datasets/datasets.py

@@ -90,7 +90,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:
@@ -192,7 +192,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:
                    pad_value = self.float_pad_value