Skip to content

Commit d1eee87

Browse files
committed
fix falcon
Signed-off-by: jiqing-feng <jiqing.feng@intel.com>
1 parent 356d51d commit d1eee87

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimum/exporters/ipex/modeling_utils.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -877,7 +877,7 @@ def forward(
877877
if hasattr(self, "linear_add_add"):
878878
output = self.linear_add_add(mlp_hidden_states, attention_output, residual)
879879
else:
880-
mlp_output = self.mlp.dense_4h_to_h(mlp_hidden_states)
880+
mlp_output = self.dense_4h_to_h(mlp_hidden_states)
881881
output = mlp_output + attention_output + residual
882882

883883
return output

0 commit comments

Comments
 (0)