From 12dd21af899de9c99d8bb0579f2fc406b8013627 Mon Sep 17 00:00:00 2001 From: bug-orz Date: Thu, 24 Jul 2025 14:02:59 +0800 Subject: [PATCH] Update cli.py --- easydistill/cli.py | 8 +------- 1 file changed, 1 insertion(+), 7 deletions(-) diff --git a/easydistill/cli.py b/easydistill/cli.py index 2287bc0..2d1f35a 100644 --- a/easydistill/cli.py +++ b/easydistill/cli.py @@ -99,10 +99,7 @@ def process(job_type, config): cmd_infer = ' '.join(cmd_infer) logging.info(f"Running command: {cmd_infer}") infer_success = run_cmd(cmd_infer) - - ############################### - infer_success=True - ############################### + if infer_success: cmd_train = [ 'accelerate', 'launch', @@ -126,9 +123,6 @@ def process(job_type, config): logging.info(f"Running command: {cmd_infer}") infer_success = run_cmd(cmd_infer) - ############################### - infer_success=True - ############################### if infer_success: cmd_train = [ 'accelerate', 'launch',