SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

Remove Longformers from ONNX-supported models (#15273)

Browse Source
This commit is contained in:
lewtun
2022-02-07 17:32:13 +01:00
committed by GitHub
parent 7a1412e12b
commit 6775b211b6
5 changed files with 3 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
utils/check_table.py
View File

@@ -210,7 +210,7 @@ def check_onnx_model_list(overwrite=False):
"""Check the model list in the serialization.mdx is consistent with the state of the lib and maybe `overwrite`."""
current_list, start_index, end_index, lines = _find_text_in_file(
filename=os.path.join(PATH_TO_DOCS, "serialization.mdx"),
start_prompt="<!--This table is automatically generated by make style, do not fill manually!-->",
start_prompt="<!--This table is automatically generated by `make fix-copies`, do not fill manually!-->",
end_prompt="The ONNX conversion is supported for the PyTorch versions of the models.",
)
new_list = get_onnx_model_list()