update CI

Signed-off-by: minmingzhu <[email protected]>
intel · Jun 19, 2024 · 4d0e247 · 4d0e247
1 parent c3cf99f
commit 4d0e247
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 7 deletions.
diff --git a/dev/scripts/patch_yaml_config.py b/dev/scripts/patch_yaml_config.py
@@ -75,12 +75,9 @@ def patch_yaml_config():
             if "finetuning_model" not in result["Training"]:
                 result["Training"]["finetuning_model"] = {}
 
-            if args.models == "mistralai/Mistral-7B-v0.1":
-                result["Dataset"]["train_file"] = "examples/data/sample_dpo_data.jsonl"
-                result["Training"]["beta"] = 0.1
-                result["Training"]["finetuning_model"]["dpo"] = True
-            else:
-                result["Training"]["finetuning_model"]["dpo"] = False
+            result["Dataset"]["train_file"] = "examples/data/sample_dpo_data.jsonl"
+            result["Training"]["beta"] = 0.1
+            result["Training"]["finetuning_model"]["dpo"] = True
 
     with open(conf_path, "w") as output:
         yaml.dump(result, output, sort_keys=False)

diff --git a/pyproject.toml b/pyproject.toml
@@ -36,7 +36,7 @@ dependencies = [
     "async_timeout",
     "typer",
     "matplotlib",
-    "trl>=0.8.6"
+    "trl==0.8.6"
 ]
 
 [project.optional-dependencies]