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

[docs] fix model checkpoint name (#36075)

Browse Source 
update model name
This commit is contained in:
Fanli Lin
2025-02-08 04:41:52 +08:00
committed by GitHub
parent ba29a439ad
commit c361b1e3d9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/source/en/serialization.md
View File

@@ -130,7 +130,7 @@ Alternative to CLI, you can export a 🤗 Transformers model to ONNX programmati
>>> from optimum.onnxruntime import ORTModelForSequenceClassification >>> from optimum.onnxruntime import ORTModelForSequenceClassification
>>> from transformers import AutoTokenizer >>> from transformers import AutoTokenizer
>>> model_checkpoint = "distilbert_base_uncased_squad" >>> model_checkpoint = "distilbert/distilbert-base-uncased-distilled-squad"
>>> save_directory = "onnx/" >>> save_directory = "onnx/"
>>> # Load a model from transformers and export it to ONNX >>> # Load a model from transformers and export it to ONNX