Efficient model for image/video generation