修复chunk_size可能传入str或list结构 (#1357)

dudulu

2024-02-05 9c5f00794e7c474d1079ff78e5c69e8c53a30e6e

 funasr/auto/auto_model.py

@@ -228,7 +228,7 @@
            data_batch = data_list[beg_idx:end_idx]
            key_batch = key_list[beg_idx:end_idx]
            batch = {"data_in": data_batch, "key": key_batch}
            if (end_idx - beg_idx) == 1 and isinstance(data_batch[0], torch.Tensor): # fbank
            if (end_idx - beg_idx) == 1 and kwargs.get("data_type", None) == "fbank": # fbank
                batch["data_in"] = data_batch[0]
                batch["data_lengths"] = input_len