From 71ef62f336c9471bf46dda92e29ea6bb012abe5a Mon Sep 17 00:00:00 2001 From: SolitaryThinker Date: Tue, 21 Jul 2026 16:05:37 -0700 Subject: [PATCH] [fix]: match Cosmos3 oracle attention contract --- fastvideo/tests/transformers/test_cosmos3_block_fingerprint.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/fastvideo/tests/transformers/test_cosmos3_block_fingerprint.py b/fastvideo/tests/transformers/test_cosmos3_block_fingerprint.py index 55ff1e929..9bbeee9b4 100644 --- a/fastvideo/tests/transformers/test_cosmos3_block_fingerprint.py +++ b/fastvideo/tests/transformers/test_cosmos3_block_fingerprint.py @@ -392,7 +392,7 @@ def _official_layer_parity( torch.cat([value_und, value_gen], dim=0), is_causal=False, ) - return from_und_gen_splits(und_output, gen_output, query_pack) + return from_und_gen_splits(und_output, gen_output, query_pack), None block.self_attn.dispatch_attention_fn = exact_dispatch und, gen, cos_und, sin_und, cos_gen, sin_gen = (