[misc] upgrade cli (#7714)

This commit is contained in:
hoshi-hiyouga
2025-04-14 15:41:22 +08:00
committed by GitHub
parent f518bfba5b
commit 7c61b35106
6 changed files with 26 additions and 10 deletions

View File

@@ -368,6 +368,7 @@ class Runner:
if args.get("deepspeed", None) is not None:
env["FORCE_TORCHRUN"] = "1"
# NOTE: DO NOT USE shell=True to avoid security risk
self.trainer = Popen(["llamafactory-cli", "train", save_cmd(args)], env=env)
yield from self.monitor()