Spaces:
Running
Running
| title: README | |
| emoji: 馃寑 | |
| colorFrom: pink | |
| colorTo: purple | |
| sdk: static | |
| pinned: false | |
| # Simillusion | |
| **Open tools for building, training and sharing world models.** | |
| - 馃 **[Simulusion](https://github.com/ParamThakkar123/simulusion)**: our PyTorch library. Dreamer, PlaNet, IRIS, DIAMOND, Genie and more behind one API. `pip install simulusion` | |
| - 馃摝 **Pretrained models**: below, each trained on a single 4 GB laptop GPU | |
| - 馃殌 **End-to-end world-model platform**: coming soon | |
| <table> | |
| <tr> | |
| <td width="33%"><a href="https://huggingface.co/simillusion/dreamer-dmc-cartpole-swingup"><img src="https://huggingface.co/simillusion/dreamer-dmc-cartpole-swingup/resolve/main/dreamer_dream.gif" alt="Dreamer" width="100%"></a></td> | |
| <td width="33%"><a href="https://huggingface.co/simillusion/iris-atari-boxing"><img src="https://huggingface.co/simillusion/iris-atari-boxing/resolve/main/iris_dream.gif" alt="IRIS" width="100%"></a></td> | |
| <td width="33%"><a href="https://huggingface.co/simillusion/planet-dmc-cartpole-swingup"><img src="https://huggingface.co/simillusion/planet-dmc-cartpole-swingup/resolve/main/planet_dream.gif" alt="PlaNet" width="100%"></a></td> | |
| </tr> | |
| </table> | |
| ```python | |
| from huggingface_hub import snapshot_download | |
| from simulusion.world_model import load_world_model | |
| wm = load_world_model(snapshot_download("simillusion/iris-atari-boxing"), "iris") | |
| ``` | |
| 馃寪 [Website](https://simulusion.vercel.app) 路 馃捇 [GitHub](https://github.com/ParamThakkar123/simulusion) 路 馃摉 [Docs](https://paramthakkar123.github.io/simulusion/) 路 馃摝 [Models](https://huggingface.co/collections/simillusion/simulusion-demo-models-6ae0419992e0177bc1695362) | |
| Tell us what would make world models useful in your work: [open an issue](https://github.com/ParamThakkar123/simulusion/issues). | |