majumderb / pabst

Code for "Unsupervised Enrichment of Persona-grounded Dialog with Background Stories", ACL 2021

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

What is variable COMPAC_MODEL?

Namo-Bang opened this issue · comments

When run 'run_pplm.py', COMPAC_MODEL(None) always raises error.
How can I modify the code?

shit codes... wasted me a day

As mentioned in the paper, you can replace COMPAC_MODEL with any language model that takes the form of GPT2. For example GPT2, DialoGPT, GPT2 trained by huggingface on PersonaChat dataset etc. Thanks.

@majumderb
Thanks! Now I'm training the compac model, and if there's a problem again, I'll leave an issue.