From 0515da673dcf67b588897b5769dba78fa5de6f24 Mon Sep 17 00:00:00 2001 From: CalamitousFelicitousness Date: Sat, 25 Apr 2026 11:47:45 +0100 Subject: [PATCH] fix(anima): handle net. prefix on custom transformer files NVIDIA/Cosmos native exports prefix every key with net., including the bundled llm_adapter half. Add it to KNOWN_PREFIXES so partition_adapter runs on stripped keys. --- pipelines/anima/anima_transformer.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pipelines/anima/anima_transformer.py b/pipelines/anima/anima_transformer.py index 0e992b690..06692dc4e 100644 --- a/pipelines/anima/anima_transformer.py +++ b/pipelines/anima/anima_transformer.py @@ -19,6 +19,7 @@ Supported input formats (safetensors only; GGUF and .pth are rejected early): - Bare BFL keys: ``blocks.0.self_attn.q_proj.weight`` (e.g. ``rdbtAnima_v027``) - ``model.diffusion_model.`` prefix (e.g. ``animaika_v35``) - ``diffusion_model.`` prefix (ComfyUI-style export) +- ``net.`` prefix (NVIDIA/Cosmos native export, e.g. ``animayume_v04``) Quantization: SDNQ (pre/post/auto) and ``layerwise_quantization`` are honored. SDNQ pre-mode is applied post-load here because this path bypasses @@ -35,7 +36,7 @@ from modules import shared, devices, sd_models, model_quant, errors from modules.logger import log -KNOWN_PREFIXES = ("model.diffusion_model.", "diffusion_model.") +KNOWN_PREFIXES = ("model.diffusion_model.", "diffusion_model.", "net.") ADAPTER_PREFIX = "llm_adapter." COSMOS_1_MARKER = "net.blocks.block1.blocks.0.block.attn.to_q.0.weight"