Reloading the model when setting the diversity level