Kevin Zhao
commited on
Commit
•
1c03163
1
Parent(s):
43e4926
Add example loading previous checkpoint
Browse files
README.md
CHANGED
@@ -6,4 +6,5 @@ datasets:
|
|
6 |
|
7 |
35M parameter baseline [GENIE](https://arxiv.org/pdf/2402.15391) for `v1.0` and `v1.1` of the [1X World Model Challenge](https://github.com/1x-technologies/1xgpt).
|
8 |
|
9 |
-
The default checkpoint corresponds to `v1.1`, but older versions can be loaded using the `revision` argument
|
|
|
|
6 |
|
7 |
35M parameter baseline [GENIE](https://arxiv.org/pdf/2402.15391) for `v1.0` and `v1.1` of the [1X World Model Challenge](https://github.com/1x-technologies/1xgpt).
|
8 |
|
9 |
+
The default checkpoint corresponds to `v1.1`, but older versions can be loaded using the `revision` argument. E.g. `STMaskGIT.from_pretrained("1x-technologies/GENIE_35M", revision="v1.0")`.
|
10 |
+
|