You could also very likely optimize the current models to be a lot more energy efficient, that'd be a win even if they didn't become more intelligent.
But none of that stops anyone else from researching or improving their models. Or a nation you've told to fuck off from doing so as well.
As has been said in the past, one doesn't put the genie back in the bottle. We only really "stopped" researching nukes because there was diminishing returns. Though one could posit we stopped because simulation became good enough or a myriad of other reasons.
When you're talking about nation state weapons capabilities I don't think you stop, but ai is even easier to share than nukes, it's just a few gigs of numbers versus heavy dangerous materials that are very difficult to source and make. Every gamer, mac owner, etc has the equiv lent of a centrifuge on their desk. Not everyone has a centrifuge on their desk.