atsr

语帆

2024-03-01 21b49bd56f44ea5921bacd8c0a3c5e35680cb405

 funasr/auto/auto_model.py

@@ -213,7 +213,7 @@
        batch_size = kwargs.get("batch_size", 1)
        # if kwargs.get("device", "cpu") == "cpu":
        #     batch_size = 1
        pdb.set_trace()

        key_list, data_list = prepare_data_iterator(input, input_len=input_len, data_type=kwargs.get("data_type", None), key=key)

        speed_stats = {}