funasr/datasets/large_datasets/utils/padding.py
@@ -6,15 +6,14 @@ def padding(data, float_pad_value=0.0, int_pad_value=-1): assert isinstance(data, list) assert "key" in data[0] assert "speech" in data[0] assert "text" in data[0] assert "speech" in data[0] or "text" in data[0] keys = [x["key"] for x in data] batch = {} data_names = data[0].keys() for data_name in data_names: if data_name == "key": if data_name == "key" or data_name =="sampling_rate": continue else: if data[0][data_name].dtype.kind == "i":
