Update modeling_florence2.py

modeling_florence2.py

@@ -2067,8 +2067,8 @@ class Florence2LanguageForConditionalGeneration(Florence2LanguagePreTrainedModel
     def __init__(self, config: Florence2LanguageConfig):
         super().__init__(config)
         self.model = Florence2LanguageModel(config)
-        self.register_buffer("final_logits_bias", torch.zeros((1, self.model.shared.num_embeddings)))
-        self.lm_head = nn.Linear(config.d_model, self.model.shared.num_embeddings, bias=False)
+        self.register_buffer("final_logits_bias", torch.zeros((1, 1)))
+        self.lm_head = nn.Linear(config.d_model, 1, bias=False)
 
         # Initialize weights and apply final processing
         self.post_init()