| from transformers import TFDistilBertForSequenceClassification | |
| tf_model = TFDistilBertForSequenceClassification.from_pretrained("distilbert/distilbert-base-uncased") | |
| Easily reuse this checkpoint for another task by switching to a different model head. |