Note 9 Pro Max Restarts frequently

After factory reset phone is getting restarted every 10 mins when it is locked. If we disable phone lock or make it never sleep then it do not restart, it works fine. But if we lock it, it will start restarting after every few minutes.

Phone details : Redmi Note 9 Pro MAX
Current Rom : MIUI 14

Please let me know any solution to this problem