What is variable COMPAC_MODEL?

majumderb / pabst

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

MIT License

10 stars 1 forks source link

What is variable COMPAC_MODEL? #3

Open Namo-Bang opened 2 years ago

Namo-Bang commented 2 years ago

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

SeaEagleI commented 2 years ago

shit codes... wasted me a day

majumderb commented 2 years ago

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.

Namo-Bang commented 2 years ago

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