There’s some conflicting stuff on the things I was reading, but it seems like the safest option would be flea treating the mother and using a flea comb on the babies. However, I would highly recommend contacting a veterinarian and getting their advice first if you can.