One risk of waiting that I've seen happen plenty of times, that you should factor in to your decision-making.
What if OM dumps her in the next 2 months???
Then you've got a depressed, in-HARD-withdrawal, still-fogged-out-and-wayward/entitled wife on your hands, SAYING she wants to come back and work on the marriage. But she's only one "I'm sorry!" phone call or text from OM1 away from crushing your heart again, and she's very vulnerable to an OM2 (it happens more than you'd think).
And you get all of that lovely stuff WITHOUT even having the benefit of having taken a strong stand that might have re-attracted your wife.