mistral-community
/

pixtral-12b

Image-Text-to-Text

Model card Files Files and versions

Himetsu commited on Sep 13, 2024

Commit

d04b7a5

·

verified ·

1 Parent(s): 4d53906

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -10,7 +10,7 @@ Transformers compatible pixtral checkpoints:
 from PIL import Image
 from transformers import AutoProcessor, LlavaForConditionalGeneration
 model_id = "Himetsu/pixtral-12b"
-model = LlavaForConditionalGeneration.from_pretrained(model_id, load_in_4bit=True)
 processor = AutoProcessor.from_pretrained(model_id)
 IMG_URLS = [

 from PIL import Image
 from transformers import AutoProcessor, LlavaForConditionalGeneration
 model_id = "Himetsu/pixtral-12b"
+model = LlavaForConditionalGeneration.from_pretrained(model_id)
 processor = AutoProcessor.from_pretrained(model_id)
 IMG_URLS = [