File tree Expand file tree Collapse file tree 1 file changed +6
-5
lines changed
Expand file tree Collapse file tree 1 file changed +6
-5
lines changed Original file line number Diff line number Diff line change @@ -171,6 +171,10 @@ def tokenize_messages(
171171 new_line_token_id = self .encode ("\n " , add_bos = False , add_eos = False )
172172
173173 for message in messages :
174+ # Skip system prompt
175+ if ignore_system_prompts and message .role == "system" :
176+ continue
177+
174178 # Prepend BOS on start of new turns
175179 if start_of_turn :
176180 tokenized_messages .append (self .bos_id )
@@ -186,11 +190,8 @@ def tokenize_messages(
186190 end_of_turn = True
187191 mask .append (message .masked )
188192 elif message .role == "system" :
189- if ignore_system_prompts :
190- continue
191- else :
192- tokenized_messages .append (self .special_tokens ["<|system|>" ])
193- mask .append (message .masked )
193+ tokenized_messages .append (self .special_tokens ["<|system|>" ])
194+ mask .append (message .masked )
194195 else :
195196 raise ValueError (
196197 f"Unknown role '{ message .role } ' for message: '{ message .content } '"
You can’t perform that action at this time.
0 commit comments