From 5bc49ba294e67cbf81a83731e591e5abd5f29d19 Mon Sep 17 00:00:00 2001 From: mingyi456 Date: Sat, 3 Oct 2026 23:21:27 +0800 Subject: [PATCH 1/2] Fix single file loading for Anima models with `model.diffusion_model.` prefix instead of `.net` prefix, and remove unused `llm_adapter` keys --- src/diffusers/loaders/single_file_utils.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/diffusers/loaders/single_file_utils.py b/src/diffusers/loaders/single_file_utils.py index 3e160be02b4b..366d838b9695 100644 --- a/src/diffusers/loaders/single_file_utils.py +++ b/src/diffusers/loaders/single_file_utils.py @@ -3723,6 +3723,13 @@ def swap_scale_shift(weight): def convert_cosmos_transformer_checkpoint_to_diffusers(checkpoint, **kwargs): + + for k in list(checkpoint.keys()): + if "llm_adapter" in k: + _ = checkpoint.pop(k) + else: + checkpoint[k.removeprefix("model.diffusion_model.")] = checkpoint.pop(k) + converted_state_dict = {key: checkpoint.pop(key) for key in list(checkpoint.keys())} def remove_keys_(key: str, state_dict): From 45ade8668ad2733a2f79f64642deba05b16efe1a Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" Date: Fri, 23 Oct 2026 12:06:48 +0000 Subject: [PATCH 2/2] Apply style fixes --- src/diffusers/loaders/__init__.py | 2 +- src/diffusers/loaders/single_file_utils.py | 1 - 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/src/diffusers/loaders/__init__.py b/src/diffusers/loaders/__init__.py index 868e70e48736..764e5fda68b7 100644 --- a/src/diffusers/loaders/__init__.py +++ b/src/diffusers/loaders/__init__.py @@ -7,12 +7,12 @@ _import_structure = {} if is_torch_available(): + _import_structure["lora_pipeline"] = ["RFD4ProteinaLoraLoaderMixin"] _import_structure["single_file_model"] = ["FromOriginalModelMixin"] _import_structure["transformer_flux"] = ["FluxTransformer2DLoadersMixin"] _import_structure["transformer_sd3"] = ["SD3Transformer2DLoadersMixin"] _import_structure["unet"] = ["UNet2DConditionLoadersMixin"] _import_structure["utils"] = ["AttnProcsLayers"] - _import_structure["lora_pipeline"] = ["RFD4ProteinaLoraLoaderMixin"] if is_transformers_available(): _import_structure["single_file"] = ["FromSingleFileMixin"] _import_structure["lora_pipeline"] += [ diff --git a/src/diffusers/loaders/single_file_utils.py b/src/diffusers/loaders/single_file_utils.py index 63ebd13c2002..3f1e21d4eb01 100644 --- a/src/diffusers/loaders/single_file_utils.py +++ b/src/diffusers/loaders/single_file_utils.py @@ -3737,7 +3737,6 @@ def swap_scale_shift(weight): def convert_cosmos_transformer_checkpoint_to_diffusers(checkpoint, **kwargs): - for k in list(checkpoint.keys()): if "llm_adapter" in k: _ = checkpoint.pop(k)