Add GenerationMixin to ConditionalGeneration class

modeling_florence2.py

@@ -28,6 +28,7 @@ from collections import OrderedDict
 from einops import rearrange
 from timm.models.layers import DropPath, trunc_normal_
 
+from transformers import GenerationMixin
 from transformers.modeling_utils import PreTrainedModel
 from transformers.utils import (
     ModelOutput,
@@ -2529,7 +2530,7 @@ class Florence2VisionModelWithProjection(Florence2PreTrainedModel):
     """The FLORENCE2 model which consists of a vision backbone and a language model.""",
     FLORENCE2_START_DOCSTRING,
 )
-class Florence2ForConditionalGeneration(Florence2PreTrainedModel):
+class Florence2ForConditionalGeneration(Florence2PreTrainedModel, GenerationMixin):
     def __init__(self, config: Florence2Config):
         super().__init__(config)
         assert config.vision_config.model_type == 'davit', 'only DaViT is supported for now'