Skip to content

Commit 7bab46c

Browse files
committed
filter for beta and eps only, otherwise there is also the sincos cache buffer
Signed-off-by: Nathan Ranchin <[email protected]>
1 parent 3abf386 commit 7bab46c

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

vllm/model_executor/models/apertus.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -418,7 +418,8 @@ def load_weights(self, weights: Iterable[tuple[str,
418418

419419
# we need to load the buffers for beta and eps (XIELU)
420420
for name, buffer in self.named_buffers():
421-
params_dict[name] = buffer
421+
if name.endswith(".beta") or name.endswith(".eps"):
422+
params_dict[name] = buffer
422423

423424
loaded_params: set[str] = set()
424425
for name, loaded_weight in weights:

0 commit comments

Comments
 (0)