Unverified Commit 4fed29e3 authored by amyeroberts's avatar amyeroberts Committed by GitHub
Browse files

Fix SigLip classification doctest (#30475)

* Fix SigLip classification doctest

* Remove extra line

* Update src/transformers/models/siglip/modeling_siglip.py
parent 30ee508c
...@@ -1258,7 +1258,7 @@ class SiglipForImageClassification(SiglipPreTrainedModel): ...@@ -1258,7 +1258,7 @@ class SiglipForImageClassification(SiglipPreTrainedModel):
>>> # model predicts one of the two classes >>> # model predicts one of the two classes
>>> predicted_class_idx = logits.argmax(-1).item() >>> predicted_class_idx = logits.argmax(-1).item()
>>> print("Predicted class:", model.config.id2label[predicted_class_idx]) >>> print("Predicted class:", model.config.id2label[predicted_class_idx])
Predicted class: LABEL_0 Predicted class: LABEL_1
```""" ```"""
output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions output_attentions = output_attentions if output_attentions is not None else self.config.output_attentions
output_hidden_states = ( output_hidden_states = (
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment