1x-technologies/world_model_tokenized_data
Updated • 285 • 31
How to use 1x-technologies/GENIE_35M with Transformers:
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("1x-technologies/GENIE_35M", dtype="auto")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").