llama: fix non-local x path (#15580)

2026-08-14 02:21:26 +10:00 · 2026-08-14 02:21:26 +10:00 · af3d2153a7
parent e535e59e13
commit af3d2153a7
1 changed files with 2 additions and 1 deletions
--- a/comfy/text_encoders/llama.py
+++ b/comfy/text_encoders/llama.py
@ -906,7 +906,8 @@ class Llama2_(nn.Module):
                past_kv.prepare(seq_len)

            def core():
-                _, current_kv = layer(
+                nonlocal x
+                x, current_kv = layer(
                    x=x,
                    attention_mask=mask,
                    freqs_cis=freqs_cis,