From d65e1b960cf88916ca14fa27528149579038ac68 Mon Sep 17 00:00:00 2001 From: NanoCode012 Date: Fri, 6 Mar 2026 23:45:03 +0700 Subject: [PATCH] fix: add guard for _initialize_missing_keys patch (#3469) [skip ci] --- src/axolotl/monkeypatch/accelerate/fsdp2.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/src/axolotl/monkeypatch/accelerate/fsdp2.py b/src/axolotl/monkeypatch/accelerate/fsdp2.py index 1fa589f07..87b537655 100644 --- a/src/axolotl/monkeypatch/accelerate/fsdp2.py +++ b/src/axolotl/monkeypatch/accelerate/fsdp2.py @@ -498,6 +498,9 @@ def patch_initialize_missing_keys_for_fsdp(): from transformers import PreTrainedModel from transformers.modeling_utils import is_fsdp_enabled, is_local_dist_rank_0 + if getattr(PreTrainedModel._initialize_missing_keys, "_axolotl_patched", False): + return + _original_initialize_missing_keys = PreTrainedModel._initialize_missing_keys def _patched_initialize_missing_keys(self, is_quantized: bool) -> None: @@ -510,6 +513,7 @@ def patch_initialize_missing_keys_for_fsdp(): _original_initialize_missing_keys(self, is_quantized) PreTrainedModel._initialize_missing_keys = _patched_initialize_missing_keys + PreTrainedModel._initialize_missing_keys._axolotl_patched = True def patch_accelerate_fsdp2():