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