How to save and load the custom Hugging face model including config.json file using pytorch

[shivaays]

HI @pchhapolika ,

I am looking for the similar approach with XLM roberta model, I got my custom trained model, while saving the custom layers are not getting saved, Please let me know if you found the approach.