AFAIK you can run either 2wd or 4wd pump, but use the appropriate thermostat housing for the pump you choose, although this will probably be dictated by which head you have, as the thermo housing fixings are different.
I suspect it'll be easier to use a 2wd pump if you have a 2wd head and thermostat housing (and a LOT cheaper )