Skip to content

Commit 3725600

Browse files
authored
Merge pull request #164 from xuanyuanking/add_dolly_as_input_model
Add Dolly as the Input Model
2 parents eb8f146 + a28d031 commit 3725600

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

training/consts.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,9 @@
44
"EleutherAI/pythia-6.9b",
55
"EleutherAI/pythia-12b",
66
"EleutherAI/gpt-j-6B",
7+
"databricks/dolly-v2-3b",
8+
"databricks/dolly-v2-7b",
9+
"databricks/dolly-v2-12b"
710
]
811
DEFAULT_TRAINING_DATASET = "databricks/databricks-dolly-15k"
912
INTRO_BLURB = (

0 commit comments

Comments
 (0)