microsoft

/

Florence-2-base

like 178

Follow

Microsoft 5,128

Related to florence2-large discussion.

Florence-2 currently triggers the following deprecation warning in the transformers library:

Florence2LanguageForConditionalGeneration has generative capabilities, as `prepare_inputs_for_generation` is explicitly overwritten. However, it doesn't directly inherit from `GenerationMixin`. From 👉v4.50👈 onwards, `PreTrainedModel` will NOT inherit from `GenerationMixin`, and this model will lose the ability to call `generate` and other related functions.
  - If you're using `trust_remote_code=True`, you can get rid of this warning by loading the model with an auto class. See https://huggingface.co/docs/transformers/en/model_doc/auto#auto-classes
  - If you are the owner of the model architecture code, please modify your model class such that it inherits from `GenerationMixin` (after `PreTrainedModel`, otherwise you'll get an exception).
 - If you are not the owner of the model architecture class, please contact the model code owner to update it.

This PR follows the advice of the warning and adds GenerationMixin as a parent class of Florence2LanguageForConditionalGeneration.

cc: @leoxiaobin @nielsr