[Python] How do you save a torch compiled hugging face pre trained model in a lightning module?

I have a pytorch lightning module in which: self.model = torch.compile(self.model) Eventually, I would like to save this pretrained model using .save_pretrained(). What is the proper way to do that?

Continue reading...