Add get_output_embeddings method
#10
by
xray1111
- opened
- modeling_florence2.py +3 -0
modeling_florence2.py
CHANGED
@@ -2588,6 +2588,9 @@ class Florence2ForConditionalGeneration(Florence2PreTrainedModel):
|
|
2588 |
def get_input_embeddings(self):
|
2589 |
return self.language_model.get_input_embeddings()
|
2590 |
|
|
|
|
|
|
|
2591 |
def resize_token_embeddings(self, new_num_tokens: Optional[int] = None, pad_to_multiple_of=None) -> nn.Embedding:
|
2592 |
model_embeds = self.language_model.resize_token_embeddings(new_num_tokens, pad_to_multiple_of)
|
2593 |
# update vocab size
|
|
|
2588 |
def get_input_embeddings(self):
|
2589 |
return self.language_model.get_input_embeddings()
|
2590 |
|
2591 |
+
def get_output_embeddings(self):
|
2592 |
+
return self.language_model.get_output_embeddings()
|
2593 |
+
|
2594 |
def resize_token_embeddings(self, new_num_tokens: Optional[int] = None, pad_to_multiple_of=None) -> nn.Embedding:
|
2595 |
model_embeds = self.language_model.resize_token_embeddings(new_num_tokens, pad_to_multiple_of)
|
2596 |
# update vocab size
|