Skip to content

Commit

Permalink
Adding "longrope" for Phi-3 (#2172) (#2179)
Browse files Browse the repository at this point in the history
Adding "longrope" for phi-3
  • Loading branch information
amihalik authored and ErikKaum committed Jul 26, 2024
1 parent 348b3b4 commit f598d24
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion server/text_generation_server/layers/rotary.py
Original file line number Diff line number Diff line change
Expand Up @@ -110,7 +110,7 @@ def static(cls, config, dim, base, device):
beta_fast=32,
beta_slow=1,
)
elif rope_scaling["type"] == "su":
elif rope_scaling["type"] in ["su", "longrope"]:
short_factor = torch.tensor(
rope_scaling["short_factor"], dtype=torch.float32, device=device
)
Expand Down

0 comments on commit f598d24

Please sign in to comment.