I published a repo that shows how you can extend the existing HF DLCs in order to use transformers versions higher than 4.17.0.
The repo shows how to extend the Inference Container, but the same applies to the Training DLC. Hope that helps!
1 Like