From 59f3fd35f1ec52e21223792b7adceb6c31ac3b43 Mon Sep 17 00:00:00 2001 From: greycooker <526929599@qq.com> Date: Wed, 3 Jan 2024 09:02:51 +0000 Subject: [PATCH] create new pull request --- llm/finetune_generation.py | 1 - tests/test_argparser.py | 4 ---- 2 files changed, 5 deletions(-) diff --git a/llm/finetune_generation.py b/llm/finetune_generation.py index 605a0cfe098a..c0add5ee29d6 100644 --- a/llm/finetune_generation.py +++ b/llm/finetune_generation.py @@ -68,7 +68,6 @@ def main(): gen_args, quant_args, model_args, data_args, training_args = parser.parse_json_file_and_cmd_lines( json_file_idx ) - training_args.print_config(model_args, "Model") training_args.print_config(data_args, "Data") training_args.print_config(quant_args, "Quant") diff --git a/tests/test_argparser.py b/tests/test_argparser.py index a5b881168d98..46d557ce46f5 100644 --- a/tests/test_argparser.py +++ b/tests/test_argparser.py @@ -31,7 +31,6 @@ def parse_args(): model_args = parser.parse_args_into_dataclasses() else: json_file_idx = json_indices[0] - # pdb.set_trace() model_args = parser.parse_json_file_and_cmd_lines(json_file_idx) return model_args @@ -86,7 +85,6 @@ def test_parse_json_file(self): with tempfile.NamedTemporaryFile(mode="w", suffix=".json", delete=False) as tmpfile: create_json_from_dict(ArgparserTest.args_dict, tmpfile.name) tmpfile_path = tmpfile.name - # pdb.set_trace() with patch("sys.argv", [ArgparserTest.script_name, tmpfile_path]): model_args = vars(parse_args()[0]) for key, value in ArgparserTest.args_dict.items(): @@ -113,7 +111,6 @@ def test_parse_json_file_and_cmd_lines_with_conflict(self): with tempfile.NamedTemporaryFile(mode="w", suffix=".json", delete=False) as tmpfile: json.dump(ArgparserTest.args_dict, tmpfile) tmpfile_path = tmpfile.name - # 构造命令行参数,包含与 JSON 文件中的参数冲突的值 cmd_line_args = [ ArgparserTest.script_name, tmpfile_path, @@ -124,7 +121,6 @@ def test_parse_json_file_and_cmd_lines_with_conflict(self): "--log_on_each_node", "False", ] - # 测试解析 with patch("sys.argv", cmd_line_args): model_args = vars(parse_args()[0]) self.assertEqual(model_args.get("min_learning_rate"), 2e-5)