Atfer Divorce on non imm o based on marriage

Hi, I will soon be entering on a 90 day non immigrant o visa based on marriage to a thai national. We intend to divorce while we are in Thailand and I’d like to know if my visa becomes invalid as soon as we divorce or would I be allowed to stay until the 90 days is up. Thankyou.