维石
2024-06-03 c0e5107b3924c01867749e826332faa058bd02c6
funasr/auto/auto_model.py
@@ -580,12 +580,6 @@
        )
        with torch.no_grad():
            if type == "onnx":
                export_dir = export_utils.export_onnx(model=model, data_in=data_list, **kwargs)
            else:
                export_dir = export_utils.export_torchscripts(
                    model=model, data_in=data_list, **kwargs
                )
            export_dir = export_utils.export(model=model, data_in=data_list,  **kwargs)
        return export_dir