I personally don't use treats to train puppies for my own personal reasons, but what about some cheese cut into little tiny cubes? (Just make sure you switch it up because too much cheese isn't great for them.)
Or you could use carrots. Sliced up, these make a great healthy snack for dogs.
