improved generation behavior when using models