emotion2vec

游雁

2024-12-17 fdafd3f6bc2f04d16e7cab5afcdb1257e87a8a78

 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)