diff --git a/examples/models/llama/static_attention.py b/examples/models/llama/static_attention.py index b42371dc090..95bae1b766a 100644 --- a/examples/models/llama/static_attention.py +++ b/examples/models/llama/static_attention.py @@ -259,7 +259,7 @@ def __init__( } rope = Rope(config) - freqs = rope.get_freqs(None, config.max_seq_len) + freqs = rope.get_freqs(None, config.max_context_len) self.freqs_cos = freqs[0].to(dtype) self.freqs_sin = freqs[1].to(dtype)