Diffusion limit for the partner model at its critical value