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
36
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