v1.1.17

游雁

2024-12-11 0f3d2d1266fec1b1666be1a4e0a2d234eb152567

 funasr/utils/export_utils.py

@@ -65,8 +65,9 @@
    **kwargs,
):

    device = kwargs.get("device", "cpu")
    dummy_input = model.export_dummy_inputs()
    dummy_input = (dummy_input[0].to("cuda"), dummy_input[1].to("cuda"))
    dummy_input = (dummy_input[0].to(device), dummy_input[1].to(device))


    verbose = kwargs.get("verbose", False)