File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
vllm/model_executor/models Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -108,7 +108,7 @@ def forward(self,
108
108
bucket_coords_w ).flatten ()
109
109
position_ids [batch_idx ][p_attn_mask .view (- 1 ).cpu ()] = pos_ids
110
110
position_ids = position_ids .to (self .position_embedding .weight .device )
111
- embeddings = embeddings + self .position_embedding (position_ids )
111
+ embeddings += self .position_embedding (position_ids )
112
112
return embeddings
113
113
114
114
@@ -262,11 +262,11 @@ def forward(
262
262
residual = hidden_states
263
263
hidden_states = self .layer_norm1 (hidden_states )
264
264
hidden_states = self .self_attn (hidden_states )
265
- hidden_states = residual + hidden_states
265
+ hidden_states + = residual
266
266
residual = hidden_states
267
267
hidden_states = self .layer_norm2 (hidden_states )
268
268
hidden_states = self .mlp (hidden_states )
269
- hidden_states = residual + hidden_states
269
+ hidden_states + = residual
270
270
return hidden_states
271
271
272
272
You can’t perform that action at this time.
0 commit comments