don't fail if can't cast weights due to offload when merging (#1172) …

…[skip ci]
axolotl-ai-cloud · Jan 23, 2024 · fb7f9b9 · fb7f9b9
1 parent cc25039
commit fb7f9b9
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/axolotl/cli/__init__.py b/src/axolotl/cli/__init__.py
@@ -78,7 +78,10 @@ def do_merge_lora(
 
     LOG.info("running merge of LoRA with base model")
     model = model.merge_and_unload(progressbar=True)
-    model.to(dtype=cfg.torch_dtype)
+    try:
+        model.to(dtype=cfg.torch_dtype)
+    except RuntimeError:
+        pass
 
     if cfg.local_rank == 0:
         LOG.info(f"saving merged model to: {str(Path(cfg.output_dir) / 'merged')}")