From c361b1e3d9c106ee63dbe9b1ffecd0b31d221cc4 Mon Sep 17 00:00:00 2001 From: Fanli Lin Date: Sat, 8 Feb 2025 04:41:52 +0800 Subject: [PATCH] [docs] fix model checkpoint name (#36075) update model name --- docs/source/en/serialization.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/en/serialization.md b/docs/source/en/serialization.md index eacda34f7..158db9288 100644 --- a/docs/source/en/serialization.md +++ b/docs/source/en/serialization.md @@ -130,7 +130,7 @@ Alternative to CLI, you can export a 🤗 Transformers model to ONNX programmati >>> from optimum.onnxruntime import ORTModelForSequenceClassification >>> from transformers import AutoTokenizer ->>> model_checkpoint = "distilbert_base_uncased_squad" +>>> model_checkpoint = "distilbert/distilbert-base-uncased-distilled-squad" >>> save_directory = "onnx/" >>> # Load a model from transformers and export it to ONNX