From 5d3374161bdd9063641e52efdbea21b91e9042f6 Mon Sep 17 00:00:00 2001 From: Amy Roberts <22614925+amyeroberts@users.noreply.github.com> Date: Fri, 20 Sep 2024 18:47:17 +0100 Subject: [PATCH] Update pixtral example checkpoint --- docs/source/en/model_doc/pixtral.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/en/model_doc/pixtral.md b/docs/source/en/model_doc/pixtral.md index 1c610d19b681c4..b4b3fdbbfff684 100644 --- a/docs/source/en/model_doc/pixtral.md +++ b/docs/source/en/model_doc/pixtral.md @@ -39,7 +39,7 @@ Here is an example of how to run it: from transformers import LlavaForConditionalGeneration, AutoProcessor from PIL import Image -model_id = "hf-internal-testing/pixtral-12b" +model_id = "mistral-community/pixtral-12b" model = LlavaForConditionalGeneration.from_pretrained(model_id).to("cuda") processor = AutoProcessor.from_pretrained(model_id)