Faced Similar Issue. Fixed it by the following steps

1.Increase the lambda timeout
2.Add ALB and use it as target group lambda

This should solve the timeout as it bypasses api gateway.
some notes
1.When targeting use alias otherwise you might face 502 bad gateway after update