Diffusion in hydrodynamic models of stratified fluids