DeepBeepMeep
/

LTX_Video

Diffusion Single File

Model card Files Files and versions

DeepBeepMeep commited on May 18, 2025

Commit

31aebaf

·

verified ·

1 Parent(s): 13c8fff

Upload modeling_florence2.py

Files changed (1) hide show

Florence2/modeling_florence2.py +2 -1

Florence2/modeling_florence2.py CHANGED Viewed

@@ -2058,8 +2058,9 @@ class Florence2LanguageModel(Florence2LanguagePreTrainedModel):
             encoder_attentions=encoder_outputs.attentions,
         )
-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"]

             encoder_attentions=encoder_outputs.attentions,
         )
+from transformers.generation.utils import GenerationMixin
+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"]