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
change
sglang
Commits
8b6a4486
Unverified
Commit
8b6a4486
authored
Jan 19, 2025
by
giorgiopiatti-dfinity
Committed by
GitHub
Jan 19, 2025
Browse files
fix missing revision arg when loading tokenizer (#2982)
parent
a69cb5cf
Changes
5
Hide whitespace changes
Inline
Side-by-side
Showing
5 changed files
with
8 additions
and
0 deletions
+8
-0
python/sglang/srt/managers/detokenizer_manager.py
python/sglang/srt/managers/detokenizer_manager.py
+1
-0
python/sglang/srt/managers/scheduler.py
python/sglang/srt/managers/scheduler.py
+2
-0
python/sglang/srt/managers/tokenizer_manager.py
python/sglang/srt/managers/tokenizer_manager.py
+2
-0
python/sglang/srt/managers/tp_worker.py
python/sglang/srt/managers/tp_worker.py
+2
-0
python/sglang/srt/server.py
python/sglang/srt/server.py
+1
-0
No files found.
python/sglang/srt/managers/detokenizer_manager.py
View file @
8b6a4486
...
...
@@ -71,6 +71,7 @@ class DetokenizerManager:
server_args
.
tokenizer_path
,
tokenizer_mode
=
server_args
.
tokenizer_mode
,
trust_remote_code
=
server_args
.
trust_remote_code
,
revision
=
server_args
.
revision
,
)
self
.
decode_status
=
LimitedCapacityDict
()
...
...
python/sglang/srt/managers/scheduler.py
View file @
8b6a4486
...
...
@@ -206,6 +206,7 @@ class Scheduler:
server_args
.
tokenizer_path
,
tokenizer_mode
=
server_args
.
tokenizer_mode
,
trust_remote_code
=
server_args
.
trust_remote_code
,
revision
=
server_args
.
revision
,
)
self
.
tokenizer
=
self
.
processor
.
tokenizer
else
:
...
...
@@ -213,6 +214,7 @@ class Scheduler:
server_args
.
tokenizer_path
,
tokenizer_mode
=
server_args
.
tokenizer_mode
,
trust_remote_code
=
server_args
.
trust_remote_code
,
revision
=
server_args
.
revision
,
)
# Check whether overlap can be enabled
...
...
python/sglang/srt/managers/tokenizer_manager.py
View file @
8b6a4486
...
...
@@ -158,6 +158,7 @@ class TokenizerManager:
server_args
.
tokenizer_path
,
tokenizer_mode
=
server_args
.
tokenizer_mode
,
trust_remote_code
=
server_args
.
trust_remote_code
,
revision
=
server_args
.
revision
,
)
self
.
tokenizer
=
self
.
processor
.
tokenizer
os
.
environ
[
"TOKENIZERS_PARALLELISM"
]
=
"false"
...
...
@@ -171,6 +172,7 @@ class TokenizerManager:
server_args
.
tokenizer_path
,
tokenizer_mode
=
server_args
.
tokenizer_mode
,
trust_remote_code
=
server_args
.
trust_remote_code
,
revision
=
server_args
.
revision
,
)
# Store states
...
...
python/sglang/srt/managers/tp_worker.py
View file @
8b6a4486
...
...
@@ -83,6 +83,7 @@ class TpModelWorker:
server_args
.
tokenizer_path
,
tokenizer_mode
=
server_args
.
tokenizer_mode
,
trust_remote_code
=
server_args
.
trust_remote_code
,
revision
=
server_args
.
revision
,
)
self
.
tokenizer
=
self
.
processor
.
tokenizer
else
:
...
...
@@ -90,6 +91,7 @@ class TpModelWorker:
server_args
.
tokenizer_path
,
tokenizer_mode
=
server_args
.
tokenizer_mode
,
trust_remote_code
=
server_args
.
trust_remote_code
,
revision
=
server_args
.
revision
,
)
self
.
device
=
self
.
model_runner
.
device
...
...
python/sglang/srt/server.py
View file @
8b6a4486
...
...
@@ -1027,6 +1027,7 @@ class Runtime:
self
.
server_args
.
tokenizer_path
,
tokenizer_mode
=
self
.
server_args
.
tokenizer_mode
,
trust_remote_code
=
self
.
server_args
.
trust_remote_code
,
revision
=
self
.
server_args
.
revision
,
)
async
def
async_generate
(
...
...
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