Fix incomplete outputs of FlaxBert 
# What does this PR do?

FlaxBertEncoder seems to return incomplete output (only `hidden_states`) in the case of returning a tuple.

## Who can review?

@sanchit-gandhi @patrickvonplaten 

_The documentation is not available anymore as the PR was closed or merged._