Text Generation
Transformers
PyTorch
Safetensors
English
gpt_refact
code
custom_code
Eval Results
svakhreev commited on
Commit
a95f864
·
1 Parent(s): 49f3014

Update modeling_gpt_refact.py

Browse files
Files changed (1) hide show
  1. 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}