MMI keeps turning off straight after loading

[forthegame]

2013 B8.5 Audi A4 - had it for a few years no worries, up until today in traffic I had the radio on and the MMI system turned off with no warning, no lights and black screen.

- Turned it back on

- gets past the MMI logo / booting screen

- Radio screen loads up for two seconds (the stations don't appear)

- Turns off again, no lights on the buttons either

tried turning it on a few times on the way home as well as starting the car up again and no luck.

[Steve Q]

You'll need a diagnostic check to see if any fault codes flag up. But the unit could have failed.