transformers

525dbbf8 - Remove loss from some flax models docs & examples (#15492)

Remove loss from some flax models docs & examples (#15492) * Remove return_loss from Flax models * fix more * fix Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>