diff --git a/src/axolotl/cli/merge_lora.py b/src/axolotl/cli/merge_lora.py index 611881ab1a..f3543be420 100644 --- a/src/axolotl/cli/merge_lora.py +++ b/src/axolotl/cli/merge_lora.py @@ -25,6 +25,8 @@ def do_cli(config: Path = Path("examples/"), **kwargs): load_in_8bit=False, load_in_4bit=False, flash_attention=False, + deepspeed=None, + fsdp=None, **kwargs, )