Fix model card metadata

cc4e32a verified 3 months ago

502 Bytes

license: mit
library_name: jax
tags:
  - jax
  - equinox
  - lewm
  - world-model
  - safetensors

LeWM cube JAX checkpoint

JAX-native Safetensors export of quentinll/lewm-cube for le-wm-jax.

Files:

config.json
model.safetensors

Load:

from le_wm_jax.checkpoint import load_lewm_jax

model = load_lewm_jax("DavidHidary/lewm-cube-jax")

The original PyTorch checkpoint is from the LeWM release by Lucas Maes and collaborators.