I’m using Caffeine mode when updating. You can edit the Quick Settings menu and add the Caffeine tile to have it available when needed … it prevents the screen from power saving for 5 / 10 / 30 minutes (set by repeatedly tapping on it) or indefinitely (until turned off manually or the phone rebooting).