Update README.md

README.md

@@ -116,13 +116,12 @@ def main(args):
     image_processor = vision_tower.image_processor
 
     conv = Conversation(
-        system=system='<|start_header_id|>system<|end_header_id|>\n\nAnswer the questions.',
-        roles=('<|start_header_id|>user<|end_header_id|>\n\n',
-               '<|start_header_id|>assistant<|end_header_id|>\n\n'),
+        system=system='<|system|>\nAnswer the questions.',
+        roles=('<|user|>\n', '<|assistant|>\n'),
         messages=[],
         offset=0,
         sep_style=SeparatorStyle.MPT,
-        sep='<|eot_id|>',
+        sep='<|end|>',
     )
     roles = conv.roles