changing modeling file

modeling_florence2.py

@@ -2289,6 +2289,7 @@ class Florence2Seq2SeqLMOutput(ModelOutput):
             image_hidden_states of the model produced by the vision encoder
     """
 
+    loss: 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
@@ -2757,6 +2758,8 @@ class Florence2ForConditionalGeneration(Florence2PreTrainedModel):
             output = (logits,) + outputs[1:]
             return (loss,) + output if loss is not None else output
 
+		print(loss)
+
         return Florence2Seq2SeqLMOutput(
             loss=loss,
             logits=logits,
@@ -2838,4 +2841,4 @@ class Florence2ForConditionalGeneration(Florence2PreTrainedModel):
         return self.language_model.shift_tokens_right(labels)
 
     def _reorder_cache(self, *args, **kwargs):
-        return self.language_model._reorder_cache(*args, **kwargs)
+        return self.language_model._reorder_cache(*args, **kwargs)