ViuAI commited on
Commit
4b15855
·
verified ·
1 Parent(s): 4e8045f

Update train_sft.py: DOMAIN_NAMES_V1 9 domains without safety

Browse files
Files changed (1) hide show
  1. 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 10-Domain Architecture)
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