riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-31 18:22:34 +06:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Add missing checkpoint for doctest (#21258)

Browse Source
This commit is contained in:
amyeroberts 2023-01-23 15:27:25 +00:00 committed by GitHub
parent 5603f78fc4
commit cf1a1eed70
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/transformers/models/mobilenet_v2/modeling_mobilenet_v2.py
View File

@ -637,6 +637,7 @@ class MobileNetV2ForImageClassification(MobileNetV2PreTrainedModel):
@add_start_docstrings_to_model_forward(MOBILENET_V2_INPUTS_DOCSTRING) @add_start_docstrings_to_model_forward(MOBILENET_V2_INPUTS_DOCSTRING)
@add_code_sample_docstrings( @add_code_sample_docstrings(
checkpoint=_CHECKPOINT_FOR_DOC,
output_type=ImageClassifierOutputWithNoAttention, output_type=ImageClassifierOutputWithNoAttention,
config_class=_CONFIG_FOR_DOC, config_class=_CONFIG_FOR_DOC,
expected_output=_IMAGE_CLASS_EXPECTED_OUTPUT, expected_output=_IMAGE_CLASS_EXPECTED_OUTPUT,