Enhance predict generators.