File size: 502 Bytes
397c297 cc4e32a 397c297 cc4e32a 397c297 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 | ---
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](https://github.com/davidhidary/le-wm-jax).
Files:
```text
config.json
model.safetensors
```
Load:
```python
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.
|