Himetsu commited on
Commit
77ed10f
1 Parent(s): 1eaae60

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -8,7 +8,7 @@ Transformers compatible pixtral checkpoints.
8
 
9
  How to use:
10
  ```python
11
- from transformers import AutoProcessor, AutoModelForConditionalGeneration
12
  model_id = "Himetsu/pixtral-12b"
13
  model = LlavaForConditionalGeneration.from_pretrained(model_id, load_in_4bit=True)
14
  processor = AutoProcessor.from_pretrained(model_id)
 
8
 
9
  How to use:
10
  ```python
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)