huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-06-19 15:52:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: remove check for default rope type

Browse Source
This commit is contained in:
drbh 2025-01-29 16:10:17 +00:00
parent cb7ec9cb60
commit 79550f8b47
1 changed files with 2 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/text_generation_server/layers/rotary.py
View File

@ -88,10 +88,8 @@ class PositionRotaryEmbedding(nn.Module):
rope_type = rope_scaling.get("rope_type", rope_scaling.get("type", None))
mrope_section = rope_scaling.get("mrope_section", None)
# only apply mrope if sections are provided and the rope type is mrope or default
if mrope_section is not None and (
rope_type == "mrope" or rope_type == "default"
):
# only apply mrope if sections are provided and the rope type is mrope and a section is provided
if mrope_section is not None and rope_type == "mrope":
mrope_section = rope_scaling.get("mrope_section")
return RotaryPositionEmbeddingMultimodalSections(
inv_freq, scaling_factor, mrope_section