trigger tests

huggingface · Jan 12, 2024 · c7ddba4 · c7ddba4
1 parent e9fcc94
commit c7ddba4
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/.github/workflows/slow-tests.yml b/.github/workflows/slow-tests.yml
@@ -7,6 +7,7 @@ on:
       # Run only when python files are modified
       - "trl/**.py"
       - "examples/**.py"
+      - "tests/**.py"
 env:
   RUN_SLOW: "yes"
   IS_GITHUB_CI: "1"

diff --git a/examples/scripts/dpo.py b/examples/scripts/dpo.py
@@ -47,8 +47,12 @@ class ScriptArguments:
         default=1, metadata={"help": "the number of gradient accumulation steps"}
     )
     output_dir: Optional[str] = field(default="output", metadata={"help": "the output directory"})
-    fp16: Optional[bool] = field(default=False, metadata={"help": "Whether to activate fp16 mixed precision"})
-    bf16: Optional[bool] = field(default=False, metadata={"help": "Whether to activate bf16 mixed precision"})
+    fp16: Optional[bool] = field(
+        default=False, metadata={"help": "Whether to activate fp16 mixed precision during training"}
+    )
+    bf16: Optional[bool] = field(
+        default=False, metadata={"help": "Whether to activate bf16 mixed precision during training"}
+    )
     max_length: Optional[int] = field(default=512, metadata={"help": "max length of each sample"})
     max_prompt_length: Optional[int] = field(default=128, metadata={"help": "max length of each sample's prompt"})
     max_target_length: Optional[int] = field(