Skip to content

Latest commit

 

History

History
38 lines (27 loc) · 696 Bytes

README.md

File metadata and controls

38 lines (27 loc) · 696 Bytes

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