diff --git a/docs/source/en/serialization.md b/docs/source/en/serialization.md index eacda34f711..158db928812 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