From 3724156b4dce558a1d174a7e2239f4401ad8fefd Mon Sep 17 00:00:00 2001 From: Francisco Kurucz Date: Wed, 10 Jan 2024 14:09:06 -0300 Subject: [PATCH] Fix load correct tokenizer in Mixtral model documentation (#28437) --- docs/source/en/model_doc/mixtral.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/en/model_doc/mixtral.md b/docs/source/en/model_doc/mixtral.md index 01f18bcd3c..1af3c55254 100644 --- a/docs/source/en/model_doc/mixtral.md +++ b/docs/source/en/model_doc/mixtral.md @@ -66,7 +66,7 @@ These ready-to-use checkpoints can be downloaded and used via the HuggingFace Hu >>> device = "cuda" # the device to load the model onto >>> model = AutoModelForCausalLM.from_pretrained("mistralai/Mixtral-8x7B-v0.1") ->>> tokenizer = AutoTokenizer.from_pretrained("mistralai/Mistral-8x7B") +>>> tokenizer = AutoTokenizer.from_pretrained("mistralai/Mixtral-8x7B-v0.1") >>> prompt = "My favourite condiment is"