Thomas Young
2024-12-05 46721dc84a6b4ecaac351022a47cc480c6dbd895
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)