add_confidence_score

modeling_florence2.py

@@ -29,7 +29,6 @@ from einops import rearrange
 from timm.models.layers import DropPath, trunc_normal_
 
 from transformers.modeling_utils import PreTrainedModel
-from transformers.generation.utils import GenerationMixin
 from transformers.utils import (
     ModelOutput,
     add_start_docstrings,
@@ -2060,7 +2059,7 @@ class Florence2LanguageModel(Florence2LanguagePreTrainedModel):
         )
 
 
-class Florence2LanguageForConditionalGeneration(Florence2LanguagePreTrainedModel, GenerationMixin):
+class Florence2LanguageForConditionalGeneration(Florence2LanguagePreTrainedModel):
     base_model_prefix = "model"
     _tied_weights_keys = ["encoder.embed_tokens.weight", "decoder.embed_tokens.weight", "lm_head.weight"]
     _keys_to_ignore_on_load_missing = ["final_logits_bias"]

processing_florence2.py

@@ -496,7 +496,7 @@ class CoordinatesQuantizer(object):
 
 
 class Florence2PostProcesser(object):
-    r"""
+    """
     Florence-2 post process for converting text prediction to various tasks results. 
 
     Args: