How to set different form displays for creating and editing an entity?
The media entity has already specific form handlers add and edit, so you can configure them right away.

Add both form modes to the media entity at /admin/structure/display-modes/form and enable them for the media type at /admin/structure/media/manage/image/form-display in CUSTOM DISPLAY SETTINGS.