Skip to content

Commit 9af4bf9

Browse files
committed
Remove unused function arg
Signed-off-by: Chih-Chieh-Yang <7364402+cyang49@users.noreply.github.com>
1 parent 52a1b51 commit 9af4bf9

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

vllm/model_executor/layers/mamba/ops/layernorm_gated.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -142,8 +142,7 @@ def rms_norm_gated(x,
142142
z=None,
143143
eps=1e-6,
144144
group_size=None,
145-
norm_before_gate=True,
146-
upcast=True):
145+
norm_before_gate=True):
147146
x_shape_og = x.shape
148147
# reshape input data into 2D tensor
149148
x = x.reshape(-1, x.shape[-1])

0 commit comments

Comments
 (0)