TL;DR: A new contextual world-model that helps to generalize better to new scenarios by understanding contextual factors like robot mass or strength. This contextual Dreamer outperforms existing approaches in both familiar and unfamiliar situations.