Update modeling_gpt_refact.py
Browse files- modeling_gpt_refact.py +0 -3
modeling_gpt_refact.py
CHANGED
|
@@ -512,9 +512,6 @@ class GPTRefactForCausalLM(GPTRefactPreTrainedModel):
|
|
| 512 |
# Initialize weights and apply final processing
|
| 513 |
self.post_init()
|
| 514 |
|
| 515 |
-
def get_output_embeddings(self):
|
| 516 |
-
return self.lm_head
|
| 517 |
-
|
| 518 |
def prepare_inputs_for_generation(self, input_ids, past_key_values=None, inputs_embeds=None, **kwargs):
|
| 519 |
if inputs_embeds is not None and past_key_values is None:
|
| 520 |
model_inputs = {"inputs_embeds": inputs_embeds}
|
|
|
|
| 512 |
# Initialize weights and apply final processing
|
| 513 |
self.post_init()
|
| 514 |
|
|
|
|
|
|
|
|
|
|
| 515 |
def prepare_inputs_for_generation(self, input_ids, past_key_values=None, inputs_embeds=None, **kwargs):
|
| 516 |
if inputs_embeds is not None and past_key_values is None:
|
| 517 |
model_inputs = {"inputs_embeds": inputs_embeds}
|