add prompt template class

Former-commit-id: 3d7e3a38d00aa5d9664824093043951af8c3f707
2023-06-07 11:55:25 +08:00
parent 3da427a665
commit b9feb82e4e
8 changed files with 67 additions and 40 deletions
--- a/src/utils/config.py
+++ b/src/utils/config.py
@@ -141,7 +141,7 @@ class DataTrainingArguments:
        default=0,
        metadata={"help": "Proportion of the dataset to include in the development set, should be between 0.0 and 1.0."}
    )
-    prompt_template: Optional[Literal["alpaca", "ziya"]] = field(
+    prompt_template: Optional[Literal["alpaca", "vicuna", "ziya"]] = field(
        default="alpaca",
        metadata={"help": "Which template to use for constructing prompts in training."}
    )