fix bytes

游雁

2024-12-10 92586a4a90644073f309a024705370e84470620f

 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)