From f2c38dd9dc0cc61412ad6eb97d1a66365a789fa8 Mon Sep 17 00:00:00 2001
From: comfyanonymous <121283862+comfyanonymous@users.noreply.github.com>
Date: Mon, 18 Aug 2025 20:19:06 -0400
Subject: [PATCH] Typo.

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

diff --git a/comfy/text_encoders/llama.py b/comfy/text_encoders/llama.py
index 72e35f111..3ff4ab36e 100644
--- a/comfy/text_encoders/llama.py
+++ b/comfy/text_encoders/llama.py
@@ -287,7 +287,7 @@ class Llama2_(nn.Module):
             x *= self.config.hidden_size ** 0.5
 
         if position_ids is None:
-            position_ids = torch.arange(0, x.shape[1], device=xdevice).unsqueeze(0)
+            position_ids = torch.arange(0, x.shape[1], device=x.device).unsqueeze(0)
 
         freqs_cis = precompute_freqs_cis(self.config.head_dim,
                                          position_ids,