This is the official repository for implementation of https://arxiv.org/abs/2104.04039 (Accepted at AIIDE 2021 (Oral) !!)
Plug-and-Blend introduce blending control to your continuation model and grants high-level planning capabilities to it. Great for generating stories, designed with co-creativity in mind and will benefit other PCG applications.
With the support of custom LogitsProcessor added to generate(), using P&B is easier than ever! Take a look at this colab notebook for a reference implementation of P&B LogitsProcessor that you can use with a single parameter update in your generate() !
(Hint: Don't forget to switch Runtime Type to GPU - Or generating a sentence will take a minute!)
Please consider citing this work using the following information:
@article{lin2021plug,
title={Plug-and-Blend: A Framework for Controllable Story Generation with Blended Control Codes},
author={Lin, Zhiyu and Riedl, Mark},
journal={arXiv preprint arXiv:2104.04039},
year={2021}
}