FlaxControNetModel trained with battlemaps images.