Skip to content

Commit

Permalink
Merge commit from fork
Browse files Browse the repository at this point in the history
[patch] Patch remote OS command injection vulnerability
  • Loading branch information
hiyouga authored Nov 21, 2024
2 parents c8f1998 + d20b97e commit b3aa80d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/llamafactory/webui/runner.py
Original file line number Diff line number Diff line change
Expand Up @@ -320,7 +320,7 @@ def _launch(self, data: Dict["Component", Any], do_train: bool) -> Generator[Dic
if args.get("deepspeed", None) is not None:
env["FORCE_TORCHRUN"] = "1"

self.trainer = Popen(f"llamafactory-cli train {save_cmd(args)}", env=env, shell=True)
self.trainer = Popen(["llamafactory-cli", "train", save_cmd(args)], env=env)
yield from self.monitor()

def _form_config_dict(self, data: Dict["Component", Any]) -> Dict[str, Any]:
Expand Down

0 comments on commit b3aa80d

Please sign in to comment.