From af3d2153a73f8be48719f8c77f752a454de75902 Mon Sep 17 00:00:00 2001
From: rattus <46076784+rattus128@users.noreply.github.com>
Date: Fri, 14 Aug 2026 02:21:26 +1000
Subject: [PATCH] llama: fix non-local x path (#15580)

---
 comfy/text_encoders/llama.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/comfy/text_encoders/llama.py b/comfy/text_encoders/llama.py
index 4415d6e9f..ddac547e0 100644
--- 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,