Transformers
Safetensors
GENIE_35M / README.md
Kevin Zhao
Add example loading previous checkpoint
1c03163
---
license: apache-2.0
datasets:
- 1x-technologies/worldmodel
---
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).
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")`.