Skip to content

Codebase for public release of the plug-and-blend framework.

License

Notifications You must be signed in to change notification settings

eilab-gt/plug-and-blend

 
 

Repository files navigation

plug-and-blend

This is the official repository for implementation of https://arxiv.org/abs/2104.04039 (Accepted at AIIDE 2021 (Oral) !!)

What is this?

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.

NEW!!! Plug-and-blend as LogitsProcessor

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() !

Open In Colab

If you are curious, see the demo now!

Open In Colab

(Hint: Don't forget to switch Runtime Type to GPU - Or generating a sentence will take a minute!)

If you found this work helpful

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}
}

About

Codebase for public release of the plug-and-blend framework.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 73.3%
  • Jupyter Notebook 26.7%