Update README.md
Browse files
README.md
CHANGED
|
@@ -4,8 +4,8 @@ tags:
|
|
| 4 |
- pretraining
|
| 5 |
- pixel
|
| 6 |
datasets:
|
| 7 |
-
-
|
| 8 |
-
-
|
| 9 |
language:
|
| 10 |
- en
|
| 11 |
---
|
|
@@ -43,7 +43,7 @@ Here is how to load PIXEL:
|
|
| 43 |
from pixel import PIXELConfig, PIXELForPreTraining
|
| 44 |
|
| 45 |
config = PIXELConfig.from_pretrained("Team-PIXEL/pixel-base")
|
| 46 |
-
model =
|
| 47 |
|
| 48 |
```
|
| 49 |
|
|
|
|
| 4 |
- pretraining
|
| 5 |
- pixel
|
| 6 |
datasets:
|
| 7 |
+
- Team-PIXEL/rendered-bookcorpus
|
| 8 |
+
- Team-PIXEL/rendered-wikipedia-english
|
| 9 |
language:
|
| 10 |
- en
|
| 11 |
---
|
|
|
|
| 43 |
from pixel import PIXELConfig, PIXELForPreTraining
|
| 44 |
|
| 45 |
config = PIXELConfig.from_pretrained("Team-PIXEL/pixel-base")
|
| 46 |
+
model = PIXELForPreTraining.from_pretrained("Team-PIXEL/pixel-base", config=config)
|
| 47 |
|
| 48 |
```
|
| 49 |
|