Update README.md
Browse files
README.md
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
---
|
2 |
-
license:
|
3 |
base_model:
|
4 |
- mistralai/Pixtral-12B-2409
|
5 |
library_name: transformers
|
@@ -43,8 +43,8 @@ You can install the model using the Transformers library:
|
|
43 |
from transformers import AutoProcessor, LLavaForConditionalGeneration
|
44 |
import torch
|
45 |
|
46 |
-
model = LLavaForConditionalGeneration.from_pretrained("
|
47 |
-
processor = AutoProcessor.from_pretrained("
|
48 |
```
|
49 |
|
50 |
## Example Usage
|
|
|
1 |
---
|
2 |
+
license: apache-2.0
|
3 |
base_model:
|
4 |
- mistralai/Pixtral-12B-2409
|
5 |
library_name: transformers
|
|
|
43 |
from transformers import AutoProcessor, LLavaForConditionalGeneration
|
44 |
import torch
|
45 |
|
46 |
+
model = LLavaForConditionalGeneration.from_pretrained("Prarabdha/pixtral-12b-240910-hf", torch_dtype=torch.float16, device_map="auto")
|
47 |
+
processor = AutoProcessor.from_pretrained("Prarabdha/pixtral-12b-240910-hf")
|
48 |
```
|
49 |
|
50 |
## Example Usage
|