Skip to content

GitLab

"git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "998a90bc7d8b3ecda4b1d0ef17e71afeb6729e2e"
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
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
src/transformers/models/siglip/modeling_siglip.py
View file @ 4fed29e3
...@@ -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