Hi @Katarina, sure there’s a method called AutoModelForXXX.save_pretrained that you can use: Models — transformers 4.3.0 documentation 12
You can then load the model using the AutoModelForXXX.from_pretrained method