From 5bc49ba294e67cbf81a83731e591e5abd5f29d19 Mon Sep 17 00:00:00 2001 From: mingyi456 Date: Sat, 3 Oct 2026 23:21:27 +0800 Subject: [PATCH] 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):