Welcome to HCM.
Sometimes it is nature of baby for vomiting
because do not like that typical food.Try to judge the likeness of food.
Try to engage him in toy or other activity during this time.
But it for long time than go for certain investigations to rule out any condition.
Treatment given by your doctor is on the right way.Consult pediatrician
for better medical management.