From afafb247ccfa34fe47e5c482db36627228aa384e Mon Sep 17 00:00:00 2001 From: sayakpaul Date: Sat, 14 Feb 2026 15:55:16 +0530 Subject: [PATCH] fix dduf --- src/diffusers/pipelines/transformers_loading_utils.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/diffusers/pipelines/transformers_loading_utils.py b/src/diffusers/pipelines/transformers_loading_utils.py index c750292f59..6f17c2a8cc 100644 --- a/src/diffusers/pipelines/transformers_loading_utils.py +++ b/src/diffusers/pipelines/transformers_loading_utils.py @@ -112,10 +112,14 @@ def _load_transformers_model_from_dduf( tensors = safetensors.torch.load(mmap) # Update the state dictionary with tensors state_dict.update(tensors) - return cls.from_pretrained( + model = cls.from_pretrained( pretrained_model_name_or_path=None, config=config, generation_config=generation_config, state_dict=state_dict, **kwargs, ) + # Models loaded via from_pretrained are in eval mode by default, + # but we need to preserve training mode for consistency with non-DDUF loading + model.train() + return model