From f37ffe2a70e0a48143f498f84d6ff6e15f6f081c Mon Sep 17 00:00:00 2001 From: Nathan Lambert Date: Tue, 13 Feb 2024 22:22:09 +0000 Subject: [PATCH] remove example model --- scripts/run_dpo.py | 4 ++-- scripts/run_rm.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/scripts/run_dpo.py b/scripts/run_dpo.py index 1281d8be..e0d2407b 100644 --- a/scripts/run_dpo.py +++ b/scripts/run_dpo.py @@ -43,8 +43,8 @@ def get_args(): Parse arguments strings model and chat_template """ parser = argparse.ArgumentParser() - parser.add_argument("--model", type=str, default="natolambert/gpt2-dummy-rm", help="path to model") - parser.add_argument("--ref_model", type=str, default="natolambert/gpt2-dummy-rm", help="path to model") + parser.add_argument("--model", type=str, required=True, help="path to model") + parser.add_argument("--ref_model", type=str, default=None, help="path to model") parser.add_argument("--tokenizer", type=str, default=None, help="path to non-matching tokenizer") parser.add_argument("--chat_template", type=str, default="tulu", help="path to chat template") parser.add_argument("--do_not_save", action="store_true", help="do not save results to hub (for debugging)") diff --git a/scripts/run_rm.py b/scripts/run_rm.py index 8c959414..aefd6dc3 100644 --- a/scripts/run_rm.py +++ b/scripts/run_rm.py @@ -47,7 +47,7 @@ def get_args(): Parse arguments strings model and chat_template """ parser = argparse.ArgumentParser() - parser.add_argument("--model", type=str, default="natolambert/gpt2-dummy-rm", help="path to model") + parser.add_argument("--model", type=str, required=True, help="path to model") parser.add_argument("--tokenizer", type=str, default=None, help="path to non-matching tokenizer to model") parser.add_argument("--chat_template", type=str, default="tulu", help="path to chat template") parser.add_argument(