Several different files in our typical training pipeline, such as run_diffusion.py, do not support Mega Man. In many of these cases, it's pretty simple to add support, simply by adding "MM-Simple" and "MM-Full" to each file, and updating a few file/game specific parameters.