ComfyUI/comfy/ldm/lumina
keejkrej 98bdeca914 Fix Lumina model GQA compatibility with comfy_kitchen RoPE
The comfy_kitchen apply_rope function requires xq and xk to have identical shapes, but Lumina uses Grouped Query Attention where query and key tensors can have different head counts. This adds a shape check to use apply_rope1 separately when shapes differ, maintaining compatibility with both standard attention and GQA.

🤖 Generated with [Claude Code](https://claude.com/claude-code)

Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
2026-01-06 23:52:06 +00:00
..
controlnet.py Basic implementation of z image fun control union 2.0 (#11304) 2025-12-13 01:39:11 -05:00
model.py Fix Lumina model GQA compatibility with comfy_kitchen RoPE 2026-01-06 23:52:06 +00:00