ai-models/Florence-2-large
2
0
Fork 0
mirror of https://www.modelscope.cn/AI-ModelScope/Florence-2-large.git synced 2026-04-02 21:52:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Inherit from GenerationMixin (#80)

Browse Source 
- Add GenerationMixin as parent class (c719442dd7d237edafe787824ebe245dcfa997d4)

Co-authored-by: Marvin Gazibarić <Link161@users.noreply.huggingface.co>
This commit is contained in:
ai-modelscope
2024-11-05 06:05:36 +08:00
parent 49a6a92219
commit 18c21287ed
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
modeling_florence2.py
View File

@ -29,6 +29,7 @@ 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,
@ -2059,7 +2060,7 @@ class Florence2LanguageModel(Florence2LanguagePreTrainedModel):
)
class Florence2LanguageForConditionalGeneration(Florence2LanguagePreTrainedModel):
class Florence2LanguageForConditionalGeneration(Florence2LanguagePreTrainedModel, GenerationMixin):
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"]