spit Florence 2 `collate_fn` into `train_collate_fn` and `evaluation_collate_fn`

### Description

The upcoming PaliGemma2 release introduces separate `collate_fn` for train and test splits. Let's refactor Florence2 code to make it work in the same fashion.