AI removal tool not working properly in Photoshop 25

I've just updated from Photoshop 24.7.1 to 25.0.0 on Windows 11.

Since the update, when I try to use the AI removal tool, sometimes when I highlight an area and let go of the moust the app will do nothing until I click somewhere else on the screen, This happens whether Remove After Each Stroke is ticked or not, and I'd say it happens about 30-50% of the time. If I click somewhere else, it'll draw a long pink line to that spot, and sometimes I'll have to click several times before it tries to actually do the removal. I have Use Graphics Processor ticked, and I'm using an GeForce 4080 card.