Skip to content

qkaren/COLD_decoding

Repository files navigation

Code for "COLD Decoding: Energy-based Constrained Text Generation with Langevin Dynamics"

This is the code for the following paper:

[COLD Decoding: Energy-based Constrained Text Generation with Langevin Dynamics] (https://arxiv.org/pdf/2202.11705.pdf)
Lianhui Qin, Sean Welleck, Daniel Khashabi, Yejin Choi

1) Setup Environment

pip install -r requirements.txt

2) Download this Github

git clone https://github.com/qkaren/COLD_decoding.git

3) Run Command for COLD Decoding

  • CommonGen
sh commongen.sh
  • Abductive Reasoning
sh abductive.sh
  • Counterfactual Reasoning
sh counterfactual.sh

4) Rank the generation

will add soon

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published