Update README.md
Browse files
README.md
CHANGED
|
@@ -37,8 +37,10 @@ To use our models, please first install our [`mpixel` package](https://github.co
|
|
| 37 |
from transformers import AutoModel
|
| 38 |
import mpixel.models.auto_register # registers MPIXEL with the Auto classes
|
| 39 |
from mpixel.processing import MPIXELProcessor
|
|
|
|
| 40 |
processor = MPIXELProcessor.from_pretrained("Team-PIXEL/mpixel-large")
|
| 41 |
model = AutoModel.from_pretrained("Team-PIXEL/mpixel-large")
|
|
|
|
| 42 |
inputs = processor(["Hello world!", "The quick brown fox jumps over the lazy dog."])
|
| 43 |
outputs = model(**inputs)
|
| 44 |
print(outputs.last_hidden_state.shape) # torch.Size([2, 513, 768])
|
|
|
|
| 37 |
from transformers import AutoModel
|
| 38 |
import mpixel.models.auto_register # registers MPIXEL with the Auto classes
|
| 39 |
from mpixel.processing import MPIXELProcessor
|
| 40 |
+
|
| 41 |
processor = MPIXELProcessor.from_pretrained("Team-PIXEL/mpixel-large")
|
| 42 |
model = AutoModel.from_pretrained("Team-PIXEL/mpixel-large")
|
| 43 |
+
|
| 44 |
inputs = processor(["Hello world!", "The quick brown fox jumps over the lazy dog."])
|
| 45 |
outputs = model(**inputs)
|
| 46 |
print(outputs.last_hidden_state.shape) # torch.Size([2, 513, 768])
|