How can edit the paligemma funetuning scripts to plot a validation loss curve and also convert my model to hf model

https://github.com/google-research/big_vision/blob/main/big_vision/configs/proj/paligemma/finetune_paligemma.ipynb
Can some please help ?

Start with configs/proj/paligemma/finetune_paligemma.ipynb and trace the existing fine-tuning workflow. Determine how validation loss is recorded and what conversion to a Hugging Face model should produce; done means both requested outputs are supported and demonstrated in the notebook. No tests or other entry points are mentioned.

Written by the indexing model from the issue text.