From a5d1c138fdfa8d58d5e9e6508e375f1f630675ee Mon Sep 17 00:00:00 2001
From: cameron-chen <cychen.2020@phdcs.smu.edu.sg>
Date: Wed, 29 May 2024 06:31:13 +0000
Subject: [PATCH] resolve the import error and unexpected arugment

---
 scripts/run_simpo.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/scripts/run_simpo.py b/scripts/run_simpo.py
index 77374a4..d5fabd2 100644
--- a/scripts/run_simpo.py
+++ b/scripts/run_simpo.py
@@ -26,8 +26,6 @@ from alignment import (
     DPOConfig,
     H4ArgumentParser,
     ModelArguments,
-    maybe_insert_system_message,
-    is_openai_format,
     get_checkpoint,
     get_datasets,
     get_kbit_device_map,
@@ -36,6 +34,7 @@ from alignment import (
     get_tokenizer,
     is_adapter_model,
 )
+from alignment.data import maybe_insert_system_message, is_openai_format
 from peft import PeftConfig, PeftModel
 from simpo_trainer import SimPOTrainer
 from dataclasses import dataclass, field
@@ -167,7 +166,7 @@ def main():
         splits=data_args.dataset_splits,
         configs=data_args.dataset_configs,
         columns_to_keep=["messages", "chosen", "rejected", "prompt", "completion", "label"],
-        seed=training_args.seed,
+        # seed=training_args.seed,
     )
     logger.info(
         f"Training on the following splits: {[split + ' : ' + str(dset.num_rows) for split, dset in raw_datasets.items()]}"