prepare dataset

scripts/prepare_contrain_dataset.py

@@ -155,7 +155,7 @@ def batch_iterator(name=None):
 
         for row in dataset:
             text = [
-                f"<|im_start|>{role_map[n['role']]}\n{n['value']}<|im_end|>"
+                f"<|im_start|>{role_map[n['from']]}\n{n['value']}<|im_end|>"
                 for n in row['conversations']
             ]