You might consider a pi 5. I'm using it after I installed the NVMe top mount hat. (Where I loaded Raspberry OS.) It really speeds everything up!

BTW, I'm using a M.2 2280 drive, even though the hat is designed for shorter drives. (I just tied mine down.) I installed it over the CPU fan and it works great.