From d852d7af7a73fd43f0727dcda43f4c00d2f99e29 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 23 Dec 2024 07:48:41 -0500 Subject: [PATCH] inference - don't default w accelerate, fix base model (#2216) [skip ci] --- src/axolotl/cli/main.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/axolotl/cli/main.py b/src/axolotl/cli/main.py index 6f883a2ac..14803e43b 100644 --- a/src/axolotl/cli/main.py +++ b/src/axolotl/cli/main.py @@ -93,7 +93,7 @@ def evaluate(config: str, accelerate: bool, **kwargs): @click.argument("config", type=click.Path(exists=True, path_type=str)) @click.option( "--accelerate/--no-accelerate", - default=True, + default=False, help="Use accelerate launch for multi-GPU inference", ) @click.option( @@ -124,7 +124,7 @@ def inference( if lora_model_dir: kwargs["lora_model_dir"] = lora_model_dir if base_model: - kwargs["output_dir"] = base_model + kwargs["base_model"] = base_model if accelerate: base_cmd = ["accelerate", "launch", "-m", "axolotl.cli.inference"]