Your choice of size and design of cake obviously totally depends on the style you like.  Babies tend not to eat that much cake really although there have been exceptions so it’s not necessary to get anything too big (unless you fancy cake to take home with you and I’m not ever going to discourage that!).  6-8″ is fine.