Changed inverse prompt function for training

3 years ago · a57ef238e8
parent 38a8a8eece
commit a57ef238e8
1 changed files with 1 additions and 1 deletions
--- a/prompt-learning/prompt_utils.py
+++ b/prompt-learning/prompt_utils.py
@ -17,7 +17,7 @@ PROMPT_TEMPLATES = {
        "generate": "belief states: value = $value, slot ="
    },
    "inverse-prompt": {
-        "training": INVERSE_PROMPTS["i1"],
+        "training": INVERSE_PROMPTS["i2"],
    },
    "prompt-ensemble": {
        "training": {