Unverified Commit df085d8e authored by guhur's avatar guhur Committed by GitHub
Browse files

doc: mismatch between pooler/d_output (#14641)

The model outputs a pooler_output whereas the doctype examples were using a pooled_output.
parent 0f3f045e
......@@ -717,7 +717,7 @@ class CLIPTextModel(CLIPPreTrainedModel):
>>> outputs = model(**inputs)
>>> last_hidden_state = outputs.last_hidden_state
>>> pooled_output = outputs.pooled_output # pooled (EOS token) states
>>> pooled_output = outputs.pooler_output # pooled (EOS token) states
"""
return self.text_model(
input_ids=input_ids,
......@@ -827,7 +827,7 @@ class CLIPVisionModel(CLIPPreTrainedModel):
>>> outputs = model(**inputs)
>>> last_hidden_state = outputs.last_hidden_state
>>> pooled_output = outputs.pooled_output # pooled CLS states
>>> pooled_output = outputs.pooler_output # pooled CLS states
"""
return self.vision_model(
pixel_values=pixel_values,
......
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