xiaowan0322
2024-06-06 9a9b474e7de7cc90d2ee124dc8d6c2cfa887c059
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