Kraig

Update the optimizer usage to resolve the deprecation warning from the [transformers library](https://github.com/huggingface/transformers/blob/main/src/transformers/optimization.py): "This implementation of AdamW is deprecated and will be removed in a future version. Use the PyTorch...