Merge branch 'main' of github.com:alibaba-damo-academy/FunASR merge

游雁

2024-12-17 9c0d7bee934e6ab46aa2970c7c5f34bd6031b803

 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)