xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-10 11:26:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix] fix utils.py/merge_dict func TypeError: 'type' object is not subscriptable (#3955)

Browse Source 
Co-authored-by: tianyi_zhao <tianyi.zhao@transwarp.io>
This commit is contained in:
zhaotyer 2024-04-10 12:49:11 +08:00 committed by GitHub
parent c013d32c75
commit c2e00af523
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
vllm/utils.py
View File

@ -9,7 +9,7 @@ import warnings
from collections import OrderedDict, defaultdict
from functools import lru_cache, partial
from platform import uname
from typing import (Any, Awaitable, Callable, Generic, Hashable, List,
from typing import (Any, Awaitable, Callable, Dict, Generic, Hashable, List,
Optional, Tuple, TypeVar, Union)
import psutil
@ -452,8 +452,8 @@ def maybe_expand_dim(tensor: torch.Tensor,
return tensor
def merge_dicts(dict1: dict[Any, list[Any]],
dict2: dict[Any, list[Any]]) -> dict[Any, list[Any]]:
def merge_dicts(dict1: Dict[Any, List[Any]],
dict2: Dict[Any, List[Any]]) -> Dict[Any, List[Any]]:
"""Merge 2 dicts that have key -> List of items.
When a key conflicts, the values in dict1 is prioritized.