atsr

语帆

2024-03-01 ec98a8e13859e4c3f1e55fca5f09e91be1b3d810

 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 = {}