From 62cbd8423b3b8ab173fba9cf8c7346eca5f76d32 Mon Sep 17 00:00:00 2001 From: Omar Sanseviero Date: Tue, 22 Mar 2022 21:57:19 +0100 Subject: [PATCH] Fix code repetition in serialization guide (#16346) --- docs/source/serialization.mdx | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/serialization.mdx b/docs/source/serialization.mdx index 79e96872e9..b5141b5803 100644 --- a/docs/source/serialization.mdx +++ b/docs/source/serialization.mdx @@ -169,7 +169,7 @@ checkpoint as follows: >>> from transformers import AutoTokenizer, AutoModelForSequenceClassification >>> # Load tokenizer and PyTorch weights form the Hub ->>> tokenizer = tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased") +>>> tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased") >>> pt_model = AutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased") >>> # Save to disk >>> tokenizer.save_pretrained("local-pt-checkpoint") @@ -188,7 +188,7 @@ python -m transformers.onnx --model=local-pt-checkpoint onnx/ >>> from transformers import AutoTokenizer, TFAutoModelForSequenceClassification >>> # Load tokenizer and TensorFlow weights from the Hub ->>> tokenizer = tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased") +>>> tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased") >>> tf_model = TFAutoModelForSequenceClassification.from_pretrained("distilbert-base-uncased") >>> # Save to disk >>> tokenizer.save_pretrained("local-tf-checkpoint")