Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
zhaoyu6
sglang
Commits
146f6134
"vscode:/vscode.git/clone" did not exist on "0c74471c82ed9518f66370dcc161f90d0d1da88a"
Unverified
Commit
146f6134
authored
Nov 02, 2024
by
Ran Chen
Committed by
GitHub
Nov 02, 2024
Browse files
Fix incorrect context length for llama3.2-11b (#1873)
parent
660ecb73
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
9 additions
and
5 deletions
+9
-5
python/sglang/srt/hf_transformers_utils.py
python/sglang/srt/hf_transformers_utils.py
+9
-5
No files found.
python/sglang/srt/hf_transformers_utils.py
View file @
146f6134
...
...
@@ -88,19 +88,23 @@ CONTEXT_LENGTH_KEYS = [
def
get_context_length
(
config
):
"""Get the context length of a model from a huggingface model configs."""
rope_scaling
=
getattr
(
config
,
"rope_scaling"
,
None
)
"""Get the context length of a model from a huggingface model configs.
And here the config should be text_config part if the model is a multimodal
LLM.
"""
text_config
=
getattr
(
config
,
"text_config"
,
config
)
rope_scaling
=
getattr
(
text_config
,
"rope_scaling"
,
None
)
if
rope_scaling
:
rope_scaling_factor
=
config
.
rope_scaling
.
get
(
"factor"
,
1
)
rope_scaling_factor
=
rope_scaling
.
get
(
"factor"
,
1
)
if
"original_max_position_embeddings"
in
rope_scaling
:
rope_scaling_factor
=
1
if
config
.
rope_scaling
.
get
(
"rope_type"
,
None
)
==
"llama3"
:
if
rope_scaling
.
get
(
"rope_type"
,
None
)
==
"llama3"
:
rope_scaling_factor
=
1
else
:
rope_scaling_factor
=
1
for
key
in
CONTEXT_LENGTH_KEYS
:
val
=
getattr
(
config
,
key
,
None
)
val
=
getattr
(
text_
config
,
key
,
None
)
if
val
is
not
None
:
return
int
(
rope_scaling_factor
*
val
)
return
2048
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment