RaushanA commited on
Commit
247225d
·
verified ·
1 Parent(s): d491a74

Update modeling_florence2.py

Browse files
Files changed (1) hide show
  1. modeling_florence2.py +0 -4
modeling_florence2.py CHANGED
@@ -2240,10 +2240,6 @@ class Florence2Seq2SeqLMOutput(ModelOutput):
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
 
 
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