Transformers
Safetensors
Inference Endpoints
Kevin Zhao commited on
Commit
1c03163
1 Parent(s): 43e4926

Add example loading previous checkpoint

Browse files
Files changed (1) hide show
  1. README.md +2 -1
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 of `STMaskGIT.from_pretrained()`.
 
 
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
+