Towards Mode Balancing of Generative Models via Diversity Weights