Skip to content

Commit

Permalink
fix load_weights lints
Browse files Browse the repository at this point in the history
Signed-off-by: Isotr0py <[email protected]>
  • Loading branch information
Isotr0py committed Nov 16, 2024
1 parent 2957bea commit 3c935d5
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions vllm/model_executor/models/utils.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import itertools
from dataclasses import dataclass, field
from typing import (Any, Callable, Dict, Iterable, List, Literal, Mapping,
Optional, Protocol, Tuple, Union, overload)
Optional, Protocol, Set, Tuple, Union, overload)

import torch
import torch.nn as nn
Expand Down Expand Up @@ -224,7 +224,7 @@ def load_weights(
weights: Iterable[Tuple[str, torch.Tensor]],
*,
mapper: Optional[WeightsMapper] = None,
) -> List[str]:
) -> Set[str]:
if mapper is not None:
weights = mapper.apply(weights)

Expand Down

0 comments on commit 3c935d5

Please sign in to comment.