Remove redundant configuration flags from merge cmdline

modal-labs · Feb 6, 2024 · 1abc4f8 · 1abc4f8
1 parent d1b236c
commit 1abc4f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/train.py b/src/train.py
@@ -88,7 +88,7 @@ def merge(run_folder: str):
 
         print(f"Merge from {MERGE_SRC} in {run_folder}")
 
-    MERGE_CMD = f"accelerate launch -m axolotl.cli.merge_lora ./config.yml --lora_model_dir='{MERGE_SRC}' --load_in_8bit=False --load_in_4bit=False --flash_attention=False"
+    MERGE_CMD = f"accelerate launch -m axolotl.cli.merge_lora ./config.yml --lora_model_dir='{MERGE_SRC}'"
     run_cmd(MERGE_CMD, run_folder)
 
     VOLUME_CONFIG["/runs"].commit()