Update custom_models.mdx (huggingface#16964)

BertModelForSequenceClassification -> BertForSequenceClassification
paulogrochowski · Apr 27, 2022 · cf8a7c2 · cf8a7c2
1 parent 5896b3e
commit cf8a7c2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/en/custom_models.mdx b/docs/source/en/custom_models.mdx
@@ -106,7 +106,7 @@ directly upload your config to the Hub.
 
 Now that we have our ResNet configuration, we can go on writing the model. We will actually write two: one that
 extracts the hidden features from a batch of images (like [`BertModel`]) and one that is suitable for image
-classification (like [`BertModelForSequenceClassification`]).
+classification (like [`BertForSequenceClassification`]).
 
 As we mentioned before, we'll only write a loose wrapper of the model to keep it simple for this example. The only
 thing we need to do before writing this class is a map between the block types and actual block classes. Then the