GENIE_35M / README.md

Kevin Zhao

Add example loading previous checkpoint

1c03163 4 months ago

preview code

raw

history blame contribute delete

434 Bytes

metadata

license: apache-2.0
datasets:
  - 1x-technologies/worldmodel

35M parameter baseline GENIE for v1.0 and v1.1 of the 1X World Model Challenge.

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").