Unverified Commit 9acc6e35 authored by Liangsheng Yin's avatar Liangsheng Yin Committed by GitHub
Browse files

add `.isort.cfg` (#378)

parent cf9d8efd
[settings]
profile=black
known_first_party=sglang
\ No newline at end of file
...@@ -3,11 +3,7 @@ ...@@ -3,11 +3,7 @@
import re import re
from typing import Callable, List, Optional, Union from typing import Callable, List, Optional, Union
from sglang.backend.anthropic import Anthropic
from sglang.backend.base_backend import BaseBackend 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.global_config import global_config
from sglang.lang.ir import ( from sglang.lang.ir import (
SglExpr, SglExpr,
......
from typing import List, Optional, Union
import numpy as np
from sglang.backend.base_backend import BaseBackend from sglang.backend.base_backend import BaseBackend
from sglang.lang.chat_template import get_chat_template from sglang.lang.chat_template import get_chat_template
from sglang.lang.interpreter import StreamExecutor from sglang.lang.interpreter import StreamExecutor
......
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.chat_template import get_chat_template
from sglang.lang.interpreter import StreamExecutor from sglang.lang.interpreter import StreamExecutor
......
import logging import logging
import time import time
from typing import Callable, List, Optional, Union from typing import List, Optional
import numpy as np import numpy as np
from sglang.backend.base_backend import BaseBackend from sglang.backend.base_backend import BaseBackend
from sglang.lang.chat_template import ChatTemplate, get_chat_template_by_model_path from sglang.lang.chat_template import ChatTemplate, get_chat_template_by_model_path
from sglang.lang.interpreter import StreamExecutor from sglang.lang.interpreter import StreamExecutor
......
import json import json
from typing import Callable, List, Optional, Union from typing import List, Optional
import numpy as np import numpy as np
import requests
from sglang.backend.base_backend import BaseBackend from sglang.backend.base_backend import BaseBackend
from sglang.global_config import global_config from sglang.global_config import global_config
from sglang.lang.chat_template import get_chat_template_by_model_path from sglang.lang.chat_template import get_chat_template_by_model_path
from sglang.lang.interpreter import StreamExecutor from sglang.lang.interpreter import StreamExecutor
from sglang.lang.ir import SglArgument, SglSamplingParams from sglang.lang.ir import SglSamplingParams
from sglang.utils import encode_image_base64, find_printable_text, http_request from sglang.utils import find_printable_text, http_request
class RuntimeEndpoint(BaseBackend): class RuntimeEndpoint(BaseBackend):
......
import os import os
import warnings import warnings
from typing import List, Optional, Union
import numpy as np
from sglang.backend.base_backend import BaseBackend from sglang.backend.base_backend import BaseBackend
from sglang.lang.chat_template import get_chat_template from sglang.lang.chat_template import get_chat_template
from sglang.lang.interpreter import StreamExecutor from sglang.lang.interpreter import StreamExecutor
......
from dataclasses import dataclass, field from dataclasses import dataclass
from enum import Enum, auto from enum import Enum, auto
from typing import Callable, Dict, List, Optional, Tuple from typing import Callable, Dict, List, Tuple
class ChatTemplateStyle(Enum): class ChatTemplateStyle(Enum):
......
...@@ -5,13 +5,7 @@ from typing import List, Union ...@@ -5,13 +5,7 @@ from typing import List, Union
from sglang.global_config import global_config from sglang.global_config import global_config
from sglang.lang.interpreter import ProgramState, StreamExecutor, pin_program from sglang.lang.interpreter import ProgramState, StreamExecutor, pin_program
from sglang.lang.ir import ( from sglang.lang.ir import SglArgument, SglExpr, SglSamplingParams, SglVariable
SglArgument,
SglConstantText,
SglExpr,
SglSamplingParams,
SglVariable,
)
def compile_func(function, backend): def compile_func(function, backend):
......
...@@ -7,9 +7,10 @@ import threading ...@@ -7,9 +7,10 @@ import threading
import uuid import uuid
from concurrent.futures import ThreadPoolExecutor from concurrent.futures import ThreadPoolExecutor
from contextlib import contextmanager from contextlib import contextmanager
from typing import Any, Callable, Dict, List, Optional, Union from typing import Any, Callable, Dict, List, Optional
import tqdm import tqdm
from sglang.global_config import global_config from sglang.global_config import global_config
from sglang.lang.ir import ( from sglang.lang.ir import (
SglCommitLazy, SglCommitLazy,
...@@ -17,7 +18,6 @@ from sglang.lang.ir import ( ...@@ -17,7 +18,6 @@ from sglang.lang.ir import (
SglConstantText, SglConstantText,
SglExpr, SglExpr,
SglExprList, SglExprList,
SglFunction,
SglGen, SglGen,
SglImage, SglImage,
SglRoleBegin, SglRoleBegin,
......
...@@ -472,4 +472,4 @@ class SglCommitLazy(SglExpr): ...@@ -472,4 +472,4 @@ class SglCommitLazy(SglExpr):
super().__init__() super().__init__()
def __repr__(self): def __repr__(self):
return f"CommitLazy()" return "CommitLazy()"
"""Tracing a program.""" """Tracing a program."""
import uuid 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.backend.base_backend import BaseBackend
from sglang.global_config import global_config
from sglang.lang.interpreter import ProgramState, ProgramStateGroup from sglang.lang.interpreter import ProgramState, ProgramStateGroup
from sglang.lang.ir import ( from sglang.lang.ir import (
SglArgument, SglArgument,
SglCommitLazy,
SglConcateAndAppend,
SglConstantText, SglConstantText,
SglExpr, SglExpr,
SglExprList, SglExprList,
SglFork, SglFork,
SglFunction,
SglGen, SglGen,
SglGetForkItem, SglGetForkItem,
SglRoleBegin, SglRoleBegin,
......
import interegular import interegular
from sglang.srt.constrained import FSMInfo, disk_cache, make_deterministic_fsm from sglang.srt.constrained import FSMInfo, disk_cache, make_deterministic_fsm
from sglang.srt.constrained.base_cache import BaseCache from sglang.srt.constrained.base_cache import BaseCache
......
...@@ -3,10 +3,9 @@ ...@@ -3,10 +3,9 @@
import json import json
import os import os
import warnings import warnings
from typing import List, Optional, Tuple, Union from typing import Optional, Union
from huggingface_hub import snapshot_download from huggingface_hub import snapshot_download
from sglang.srt.utils import is_multimodal_model
from transformers import ( from transformers import (
AutoConfig, AutoConfig,
AutoProcessor, AutoProcessor,
...@@ -15,6 +14,8 @@ from transformers import ( ...@@ -15,6 +14,8 @@ from transformers import (
PreTrainedTokenizerFast, PreTrainedTokenizerFast,
) )
from sglang.srt.utils import is_multimodal_model
def download_from_hf(model_path: str): def download_from_hf(model_path: str):
if os.path.exists(model_path): if os.path.exists(model_path):
......
...@@ -3,6 +3,7 @@ ...@@ -3,6 +3,7 @@
import torch import torch
import triton import triton
import triton.language as tl import triton.language as tl
from sglang.srt.utils import wrap_kernel_launcher from sglang.srt.utils import wrap_kernel_launcher
CUDA_CAPABILITY = torch.cuda.get_device_capability() CUDA_CAPABILITY = torch.cuda.get_device_capability()
......
import torch import torch
import triton import triton
import triton.language as tl import triton.language as tl
from sglang.srt.layers.context_flashattention_nopad import context_attention_fwd from sglang.srt.layers.context_flashattention_nopad import context_attention_fwd
from sglang.srt.utils import wrap_kernel_launcher from sglang.srt.utils import wrap_kernel_launcher
......
import torch import torch
from sglang.srt.managers.router.model_runner import ForwardMode, InputMetadata
from torch import nn from torch import nn
from vllm.model_executor.parallel_utils.communication_op import ( from vllm.model_executor.parallel_utils.communication_op import (
get_tensor_model_parallel_world_size, get_tensor_model_parallel_world_size,
tensor_model_parallel_all_gather, tensor_model_parallel_all_gather,
) )
from sglang.srt.managers.router.model_runner import ForwardMode, InputMetadata
class LogitsProcessor(nn.Module): class LogitsProcessor(nn.Module):
def __init__(self, config): def __init__(self, config):
......
import torch import torch
from torch import nn
from sglang.srt.layers.context_flashattention_nopad import context_attention_fwd 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.extend_attention import extend_attention_fwd
from sglang.srt.layers.token_attention import token_attention_fwd from sglang.srt.layers.token_attention import token_attention_fwd
from sglang.srt.managers.router.model_runner import ForwardMode, InputMetadata from sglang.srt.managers.router.model_runner import ForwardMode, InputMetadata
from torch import nn
class RadixAttention(nn.Module): class RadixAttention(nn.Module):
......
...@@ -4,6 +4,7 @@ ...@@ -4,6 +4,7 @@
import torch import torch
import triton import triton
import triton.language as tl import triton.language as tl
from sglang.srt.managers.router.model_runner import global_server_args_dict from sglang.srt.managers.router.model_runner import global_server_args_dict
from sglang.srt.utils import wrap_kernel_launcher from sglang.srt.utils import wrap_kernel_launcher
......
...@@ -3,6 +3,7 @@ import asyncio ...@@ -3,6 +3,7 @@ import asyncio
import uvloop import uvloop
import zmq import zmq
import zmq.asyncio import zmq.asyncio
from sglang.srt.hf_transformers_utils import get_tokenizer from sglang.srt.hf_transformers_utils import get_tokenizer
from sglang.srt.managers.io_struct import BatchStrOut, BatchTokenIDOut from sglang.srt.managers.io_struct import BatchStrOut, BatchTokenIDOut
from sglang.srt.server_args import PortArgs, ServerArgs from sglang.srt.server_args import PortArgs, ServerArgs
...@@ -83,7 +84,7 @@ def start_detokenizer_process( ...@@ -83,7 +84,7 @@ def start_detokenizer_process(
): ):
try: try:
manager = DetokenizerManager(server_args, port_args) manager = DetokenizerManager(server_args, port_args)
except Exception as e: except Exception:
pipe_writer.send(get_exception_traceback()) pipe_writer.send(get_exception_traceback())
raise raise
pipe_writer.send("init ok") pipe_writer.send("init ok")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment