Himetsu commited on
Commit
d04b7a5
1 Parent(s): 4d53906

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -10,7 +10,7 @@ Transformers compatible pixtral checkpoints:
10
  from PIL import Image
11
  from transformers import AutoProcessor, LlavaForConditionalGeneration
12
  model_id = "Himetsu/pixtral-12b"
13
- model = LlavaForConditionalGeneration.from_pretrained(model_id, load_in_4bit=True)
14
  processor = AutoProcessor.from_pretrained(model_id)
15
 
16
  IMG_URLS = [
 
10
  from PIL import Image
11
  from transformers import AutoProcessor, LlavaForConditionalGeneration
12
  model_id = "Himetsu/pixtral-12b"
13
+ model = LlavaForConditionalGeneration.from_pretrained(model_id)
14
  processor = AutoProcessor.from_pretrained(model_id)
15
 
16
  IMG_URLS = [