Integration with generation models