LeWM pusht JAX checkpoint

JAX-native Safetensors export of quentinll/lewm-pusht 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-pusht-jax")

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

Downloads last month
8
Safetensors
Model size
18M params
Tensor type
I64
ยท
F32
ยท
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support