Removed dead code

PiperOrigin-RevId: 731443798
google-ai-edge · Feb 26, 2025 · 6d58005 · 6d58005
1 parent f114cce
commit 6d58005
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/ai_edge_torch/generative/layers/attention.py b/ai_edge_torch/generative/layers/attention.py
@@ -365,7 +365,6 @@ def forward(
 
     if rope is not None:
       # Compute rotary positional embedding for query and key.
-      n_elem = int(self.config.rotary_percentage * self.config.head_dim)
       cos, sin = rope
       q, k = rotary_pos_emb.apply_rope_inline(q, k, cos, sin)