I usually put loose balls on my headsets because it's generally easier to just keep a couple of bottles of bearings that will do headsets, BBs and hubs.
I don't know if this is the correct thing to do, but I never count the balls. I just place them in the grease and keep fitting bearings until one won't fit. I remove that last one, clean it and put it back in the bottle.