File tree Expand file tree Collapse file tree 2 files changed +5
-6
lines changed
ucm/sandbox/sparse/kvcomp Expand file tree Collapse file tree 2 files changed +5
-6
lines changed Original file line number Diff line number Diff line change 3535from vllm .v1 .request import Request
3636
3737from ucm .integration .vllm .ucm_sparse .state import get_ucm_sparse
38+ from ucm .logger import init_logger
3839from ucm .sandbox .sparse .kvcomp .hash_encoder import HashEncoder
3940from ucm .sandbox .sparse .kvcomp .kvcomp_config import KvCompConfig
4041
41- from ucm .logger import init_logger
42-
4342logger = init_logger (__name__ )
4443
4544from ucm .integration .vllm .ucm_sparse .base import (
4847 UcmSparseMetadata ,
4948 UcmSparseRole ,
5049)
50+ from ucm .integration .vllm .ucm_sparse .state import get_ucm_sparse
5151from ucm .store .base import Task , UcmKVStoreBase
5252from ucm .store .factory import UcmConnectorFactory
5353
54- from ucm .integration .vllm .ucm_sparse .state import get_ucm_sparse
55-
5654
5755def stat (func ):
5856 @wraps (func )
Original file line number Diff line number Diff line change 2222THE SOFTWARE.
2323"""
2424
25- from dataclasses import dataclass , asdict , field
25+ import json
26+ from dataclasses import asdict , dataclass , field
2627from typing import List , Optional
28+
2729import numpy as np
28- import json
2930
3031from ucm .logger import init_logger
3132
You can’t perform that action at this time.
0 commit comments