I will add examples of loading a model and replacing the head as well as a pruning example here. Please let me know if there is anything else you’d like to see. Load model and replace classification head (click me to expand) Here is how to load a model and a few of the ways ...