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
9acc6e35
"projects/vscode:/vscode.git/clone" did not exist on "40fb189c2905085488d3f4a33fa233d42f3f1e9c"
Unverified
Commit
9acc6e35
authored
Apr 22, 2024
by
Liangsheng Yin
Committed by
GitHub
Apr 22, 2024
Browse files
add `.isort.cfg` (#378)
parent
cf9d8efd
Changes
57
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
30 additions
and
37 deletions
+30
-37
.isort.cfg
.isort.cfg
+3
-0
python/sglang/api.py
python/sglang/api.py
+0
-4
python/sglang/backend/anthropic.py
python/sglang/backend/anthropic.py
+0
-3
python/sglang/backend/base_backend.py
python/sglang/backend/base_backend.py
+1
-1
python/sglang/backend/openai.py
python/sglang/backend/openai.py
+2
-1
python/sglang/backend/runtime_endpoint.py
python/sglang/backend/runtime_endpoint.py
+4
-4
python/sglang/backend/vertexai.py
python/sglang/backend/vertexai.py
+0
-2
python/sglang/lang/chat_template.py
python/sglang/lang/chat_template.py
+2
-2
python/sglang/lang/compiler.py
python/sglang/lang/compiler.py
+1
-7
python/sglang/lang/interpreter.py
python/sglang/lang/interpreter.py
+2
-2
python/sglang/lang/ir.py
python/sglang/lang/ir.py
+1
-1
python/sglang/lang/tracer.py
python/sglang/lang/tracer.py
+1
-5
python/sglang/srt/constrained/jump_forward.py
python/sglang/srt/constrained/jump_forward.py
+1
-0
python/sglang/srt/hf_transformers_utils.py
python/sglang/srt/hf_transformers_utils.py
+3
-2
python/sglang/srt/layers/context_flashattention_nopad.py
python/sglang/srt/layers/context_flashattention_nopad.py
+1
-0
python/sglang/srt/layers/extend_attention.py
python/sglang/srt/layers/extend_attention.py
+1
-0
python/sglang/srt/layers/logits_processor.py
python/sglang/srt/layers/logits_processor.py
+2
-1
python/sglang/srt/layers/radix_attention.py
python/sglang/srt/layers/radix_attention.py
+2
-1
python/sglang/srt/layers/token_attention.py
python/sglang/srt/layers/token_attention.py
+1
-0
python/sglang/srt/managers/detokenizer_manager.py
python/sglang/srt/managers/detokenizer_manager.py
+2
-1
No files found.
.isort.cfg
0 → 100644
View file @
9acc6e35
[settings]
profile=black
known_first_party=sglang
\ No newline at end of file
python/sglang/api.py
View file @
9acc6e35
...
...
@@ -3,11 +3,7 @@
import
re
from
typing
import
Callable
,
List
,
Optional
,
Union
from
sglang.backend.anthropic
import
Anthropic
from
sglang.backend.base_backend
import
BaseBackend
from
sglang.backend.openai
import
OpenAI
from
sglang.backend.runtime_endpoint
import
RuntimeEndpoint
from
sglang.backend.vertexai
import
VertexAI
from
sglang.global_config
import
global_config
from
sglang.lang.ir
import
(
SglExpr
,
...
...
python/sglang/backend/anthropic.py
View file @
9acc6e35
from
typing
import
List
,
Optional
,
Union
import
numpy
as
np
from
sglang.backend.base_backend
import
BaseBackend
from
sglang.lang.chat_template
import
get_chat_template
from
sglang.lang.interpreter
import
StreamExecutor
...
...
python/sglang/backend/base_backend.py
View file @
9acc6e35
from
typing
import
Callable
,
List
,
Optional
,
Union
from
typing
import
List
,
Optional
,
Union
from
sglang.lang.chat_template
import
get_chat_template
from
sglang.lang.interpreter
import
StreamExecutor
...
...
python/sglang/backend/openai.py
View file @
9acc6e35
import
logging
import
time
from
typing
import
Callable
,
List
,
Optional
,
Union
from
typing
import
List
,
Optional
import
numpy
as
np
from
sglang.backend.base_backend
import
BaseBackend
from
sglang.lang.chat_template
import
ChatTemplate
,
get_chat_template_by_model_path
from
sglang.lang.interpreter
import
StreamExecutor
...
...
python/sglang/backend/runtime_endpoint.py
View file @
9acc6e35
import
json
from
typing
import
Callable
,
List
,
Optional
,
Union
from
typing
import
List
,
Optional
import
numpy
as
np
import
requests
from
sglang.backend.base_backend
import
BaseBackend
from
sglang.global_config
import
global_config
from
sglang.lang.chat_template
import
get_chat_template_by_model_path
from
sglang.lang.interpreter
import
StreamExecutor
from
sglang.lang.ir
import
SglArgument
,
SglSamplingParams
from
sglang.utils
import
encode_image_base64
,
find_printable_text
,
http_request
from
sglang.lang.ir
import
SglSamplingParams
from
sglang.utils
import
find_printable_text
,
http_request
class
RuntimeEndpoint
(
BaseBackend
):
...
...
python/sglang/backend/vertexai.py
View file @
9acc6e35
import
os
import
warnings
from
typing
import
List
,
Optional
,
Union
import
numpy
as
np
from
sglang.backend.base_backend
import
BaseBackend
from
sglang.lang.chat_template
import
get_chat_template
from
sglang.lang.interpreter
import
StreamExecutor
...
...
python/sglang/lang/chat_template.py
View file @
9acc6e35
from
dataclasses
import
dataclass
,
field
from
dataclasses
import
dataclass
from
enum
import
Enum
,
auto
from
typing
import
Callable
,
Dict
,
List
,
Optional
,
Tuple
from
typing
import
Callable
,
Dict
,
List
,
Tuple
class
ChatTemplateStyle
(
Enum
):
...
...
python/sglang/lang/compiler.py
View file @
9acc6e35
...
...
@@ -5,13 +5,7 @@ from typing import List, Union
from
sglang.global_config
import
global_config
from
sglang.lang.interpreter
import
ProgramState
,
StreamExecutor
,
pin_program
from
sglang.lang.ir
import
(
SglArgument
,
SglConstantText
,
SglExpr
,
SglSamplingParams
,
SglVariable
,
)
from
sglang.lang.ir
import
SglArgument
,
SglExpr
,
SglSamplingParams
,
SglVariable
def
compile_func
(
function
,
backend
):
...
...
python/sglang/lang/interpreter.py
View file @
9acc6e35
...
...
@@ -7,9 +7,10 @@ import threading
import
uuid
from
concurrent.futures
import
ThreadPoolExecutor
from
contextlib
import
contextmanager
from
typing
import
Any
,
Callable
,
Dict
,
List
,
Optional
,
Union
from
typing
import
Any
,
Callable
,
Dict
,
List
,
Optional
import
tqdm
from
sglang.global_config
import
global_config
from
sglang.lang.ir
import
(
SglCommitLazy
,
...
...
@@ -17,7 +18,6 @@ from sglang.lang.ir import (
SglConstantText
,
SglExpr
,
SglExprList
,
SglFunction
,
SglGen
,
SglImage
,
SglRoleBegin
,
...
...
python/sglang/lang/ir.py
View file @
9acc6e35
...
...
@@ -472,4 +472,4 @@ class SglCommitLazy(SglExpr):
super
().
__init__
()
def
__repr__
(
self
):
return
f
"CommitLazy()"
return
"CommitLazy()"
python/sglang/lang/tracer.py
View file @
9acc6e35
"""Tracing a program."""
import
uuid
from
typing
import
Any
,
Callable
,
Dict
,
List
,
Optional
,
Union
from
typing
import
Any
,
Dict
,
List
,
Optional
from
sglang.backend.base_backend
import
BaseBackend
from
sglang.global_config
import
global_config
from
sglang.lang.interpreter
import
ProgramState
,
ProgramStateGroup
from
sglang.lang.ir
import
(
SglArgument
,
SglCommitLazy
,
SglConcateAndAppend
,
SglConstantText
,
SglExpr
,
SglExprList
,
SglFork
,
SglFunction
,
SglGen
,
SglGetForkItem
,
SglRoleBegin
,
...
...
python/sglang/srt/constrained/jump_forward.py
View file @
9acc6e35
import
interegular
from
sglang.srt.constrained
import
FSMInfo
,
disk_cache
,
make_deterministic_fsm
from
sglang.srt.constrained.base_cache
import
BaseCache
...
...
python/sglang/srt/hf_transformers_utils.py
View file @
9acc6e35
...
...
@@ -3,10 +3,9 @@
import
json
import
os
import
warnings
from
typing
import
List
,
Optional
,
Tuple
,
Union
from
typing
import
Optional
,
Union
from
huggingface_hub
import
snapshot_download
from
sglang.srt.utils
import
is_multimodal_model
from
transformers
import
(
AutoConfig
,
AutoProcessor
,
...
...
@@ -15,6 +14,8 @@ from transformers import (
PreTrainedTokenizerFast
,
)
from
sglang.srt.utils
import
is_multimodal_model
def
download_from_hf
(
model_path
:
str
):
if
os
.
path
.
exists
(
model_path
):
...
...
python/sglang/srt/layers/context_flashattention_nopad.py
View file @
9acc6e35
...
...
@@ -3,6 +3,7 @@
import
torch
import
triton
import
triton.language
as
tl
from
sglang.srt.utils
import
wrap_kernel_launcher
CUDA_CAPABILITY
=
torch
.
cuda
.
get_device_capability
()
...
...
python/sglang/srt/layers/extend_attention.py
View file @
9acc6e35
import
torch
import
triton
import
triton.language
as
tl
from
sglang.srt.layers.context_flashattention_nopad
import
context_attention_fwd
from
sglang.srt.utils
import
wrap_kernel_launcher
...
...
python/sglang/srt/layers/logits_processor.py
View file @
9acc6e35
import
torch
from
sglang.srt.managers.router.model_runner
import
ForwardMode
,
InputMetadata
from
torch
import
nn
from
vllm.model_executor.parallel_utils.communication_op
import
(
get_tensor_model_parallel_world_size
,
tensor_model_parallel_all_gather
,
)
from
sglang.srt.managers.router.model_runner
import
ForwardMode
,
InputMetadata
class
LogitsProcessor
(
nn
.
Module
):
def
__init__
(
self
,
config
):
...
...
python/sglang/srt/layers/radix_attention.py
View file @
9acc6e35
import
torch
from
torch
import
nn
from
sglang.srt.layers.context_flashattention_nopad
import
context_attention_fwd
from
sglang.srt.layers.extend_attention
import
extend_attention_fwd
from
sglang.srt.layers.token_attention
import
token_attention_fwd
from
sglang.srt.managers.router.model_runner
import
ForwardMode
,
InputMetadata
from
torch
import
nn
class
RadixAttention
(
nn
.
Module
):
...
...
python/sglang/srt/layers/token_attention.py
View file @
9acc6e35
...
...
@@ -4,6 +4,7 @@
import
torch
import
triton
import
triton.language
as
tl
from
sglang.srt.managers.router.model_runner
import
global_server_args_dict
from
sglang.srt.utils
import
wrap_kernel_launcher
...
...
python/sglang/srt/managers/detokenizer_manager.py
View file @
9acc6e35
...
...
@@ -3,6 +3,7 @@ import asyncio
import
uvloop
import
zmq
import
zmq.asyncio
from
sglang.srt.hf_transformers_utils
import
get_tokenizer
from
sglang.srt.managers.io_struct
import
BatchStrOut
,
BatchTokenIDOut
from
sglang.srt.server_args
import
PortArgs
,
ServerArgs
...
...
@@ -83,7 +84,7 @@ def start_detokenizer_process(
):
try
:
manager
=
DetokenizerManager
(
server_args
,
port_args
)
except
Exception
as
e
:
except
Exception
:
pipe_writer
.
send
(
get_exception_traceback
())
raise
pipe_writer
.
send
(
"init ok"
)
...
...
Prev
1
2
3
Next
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