atsr

语帆

2024-03-01 fae6fd6d16e0fb060aa063790893f7555c421c4b

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