Access partial_rotary_factor from rope_parameters (#29966)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com>
This commit is contained in:
@@ -109,10 +109,7 @@ class PhiAttention(nn.Module):
|
||||
)
|
||||
|
||||
scaling = self.head_size**-0.5
|
||||
rotary_dim = int(
|
||||
config.partial_rotary_factor
|
||||
* (config.hidden_size // config.num_attention_heads)
|
||||
)
|
||||
rotary_dim = config.hidden_size // config.num_attention_heads
|
||||
assert rotary_dim % 2 == 0
|
||||
|
||||
max_position_embeddings = getattr(config, "max_position_embeddings", 2048)
|
||||
|
||||
Reference in New Issue
Block a user