changed: logits

modeling_florence2.py

@@ -2556,6 +2556,7 @@ class Florence2Seq2SeqLMOutput(ModelOutput):
     """
 
     loss: torch.FloatTensor = None
+    logits: torch.FloatTensor = None
     last_hidden_state: torch.FloatTensor = None
     past_key_values: Optional[Tuple[Tuple[torch.FloatTensor]]] = None
     decoder_hidden_states: Optional[Tuple[torch.FloatTensor, ...]] = None
@@ -3061,6 +3062,7 @@ class Florence2ForConditionalGeneration(Florence2PreTrainedModel):
             return (loss,) + output if loss is not None else output
 
         print(loss)
+        print(logits)
 
         return Florence2Seq2SeqLMOutput(
             loss=loss,