游雁
2024-06-03 1fcb7572168b65ea6b2d16de6f29413037af38db
funasr/auto/auto_frontend.py
@@ -52,7 +52,7 @@
        key_list, data_list = prepare_data_iterator(input, input_len=input_len)
        batch_size = kwargs.get("batch_size", 1)
        device = kwargs.get("device", "cpu")
        device = kwargs.get("device", "cuda")
        if device == "cpu":
            batch_size = 1