Update train_sft.py: DOMAIN_NAMES_V1 9 domains without safety
Browse files- code/train_sft.py +2 -3
code/train_sft.py
CHANGED
|
@@ -132,7 +132,7 @@ DOMAIN_NAMES_V24 = DOMAIN_NAMES_V22
|
|
| 132 |
DOMAIN_NAMES_V25 = DOMAIN_NAMES_V22
|
| 133 |
DOMAIN_NAMES_V26 = DOMAIN_NAMES_V22
|
| 134 |
|
| 135 |
-
# SFT v1 (Sarus Master
|
| 136 |
DOMAIN_NAMES_V1 = {
|
| 137 |
0: "identity_sarus",
|
| 138 |
1: "greetings_cultural",
|
|
@@ -142,8 +142,7 @@ DOMAIN_NAMES_V1 = {
|
|
| 142 |
5: "general_stem",
|
| 143 |
6: "hindi_culture",
|
| 144 |
7: "summarization_translation",
|
| 145 |
-
8: "multiturn_dialogue"
|
| 146 |
-
9: "safety_refusal"
|
| 147 |
}
|
| 148 |
DOMAIN_NAMES = DOMAIN_NAMES_V1
|
| 149 |
|
|
|
|
| 132 |
DOMAIN_NAMES_V25 = DOMAIN_NAMES_V22
|
| 133 |
DOMAIN_NAMES_V26 = DOMAIN_NAMES_V22
|
| 134 |
|
| 135 |
+
# SFT v1 (Sarus Master 9-Domain Architecture)
|
| 136 |
DOMAIN_NAMES_V1 = {
|
| 137 |
0: "identity_sarus",
|
| 138 |
1: "greetings_cultural",
|
|
|
|
| 142 |
5: "general_stem",
|
| 143 |
6: "hindi_culture",
|
| 144 |
7: "summarization_translation",
|
| 145 |
+
8: "multiturn_dialogue"
|
|
|
|
| 146 |
}
|
| 147 |
DOMAIN_NAMES = DOMAIN_NAMES_V1
|
| 148 |
|