Update modeling_florence2.py
Browse files- modeling_florence2.py +5 -1
modeling_florence2.py
CHANGED
@@ -2240,6 +2240,10 @@ class Florence2Seq2SeqLMOutput(ModelOutput):
|
|
2240 |
decoding.
|
2241 |
|
2242 |
Args:
|
|
|
|
|
|
|
|
|
2243 |
last_hidden_state (`torch.FloatTensor` of shape `(batch_size, sequence_length, hidden_size)`):
|
2244 |
Sequence of hidden-states at the output of the last layer of the decoder of the model.
|
2245 |
|
@@ -2289,6 +2293,7 @@ class Florence2Seq2SeqLMOutput(ModelOutput):
|
|
2289 |
image_hidden_states of the model produced by the vision encoder
|
2290 |
"""
|
2291 |
loss: Optional[torch.FloatTensor] = None
|
|
|
2292 |
last_hidden_state: torch.FloatTensor = None
|
2293 |
past_key_values: Optional[Tuple[Tuple[torch.FloatTensor]]] = None
|
2294 |
decoder_hidden_states: Optional[Tuple[torch.FloatTensor, ...]] = None
|
@@ -2297,7 +2302,6 @@ class Florence2Seq2SeqLMOutput(ModelOutput):
|
|
2297 |
encoder_last_hidden_state: Optional[torch.FloatTensor] = None
|
2298 |
encoder_hidden_states: Optional[Tuple[torch.FloatTensor, ...]] = None
|
2299 |
encoder_attentions: Optional[Tuple[torch.FloatTensor, ...]] = None
|
2300 |
-
logits: torch.FloatTensor = None
|
2301 |
image_hidden_states: Optional[Tuple[torch.FloatTensor, ...]] = None
|
2302 |
|
2303 |
|
|
|
2240 |
decoding.
|
2241 |
|
2242 |
Args:
|
2243 |
+
loss (`torch.FloatTensor` of shape `(1,)`, *optional*, returned when `labels` is provided):
|
2244 |
+
Language modeling loss.
|
2245 |
+
logits (`torch.FloatTensor` of shape `(batch_size, sequence_length, config.vocab_size)`):
|
2246 |
+
Prediction scores of the language modeling head (scores for each vocabulary token before SoftMax).
|
2247 |
last_hidden_state (`torch.FloatTensor` of shape `(batch_size, sequence_length, hidden_size)`):
|
2248 |
Sequence of hidden-states at the output of the last layer of the decoder of the model.
|
2249 |
|
|
|
2293 |
image_hidden_states of the model produced by the vision encoder
|
2294 |
"""
|
2295 |
loss: Optional[torch.FloatTensor] = None
|
2296 |
+
logits: torch.FloatTensor = None
|
2297 |
last_hidden_state: torch.FloatTensor = None
|
2298 |
past_key_values: Optional[Tuple[Tuple[torch.FloatTensor]]] = None
|
2299 |
decoder_hidden_states: Optional[Tuple[torch.FloatTensor, ...]] = None
|
|
|
2302 |
encoder_last_hidden_state: Optional[torch.FloatTensor] = None
|
2303 |
encoder_hidden_states: Optional[Tuple[torch.FloatTensor, ...]] = None
|
2304 |
encoder_attentions: Optional[Tuple[torch.FloatTensor, ...]] = None
|
|
|
2305 |
image_hidden_states: Optional[Tuple[torch.FloatTensor, ...]] = None
|
2306 |
|
2307 |
|