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

Update modeling_gpt_refact.py

Browse files
Files changed (1) hide show
  1. modeling_gpt_refact.py +0 -6
modeling_gpt_refact.py CHANGED
@@ -372,9 +372,6 @@ class GPTRefactModel(GPTRefactPreTrainedModel):
372
  def get_input_embeddings(self):
373
  return self.wte
374
 
375
- def set_input_embeddings(self, new_embeddings):
376
- self.wte = new_embeddings
377
-
378
  def forward(
379
  self,
380
  input_ids: Optional[torch.Tensor] = None,
@@ -518,9 +515,6 @@ class GPTRefactForCausalLM(GPTRefactPreTrainedModel):
518
  def get_output_embeddings(self):
519
  return self.lm_head
520
 
521
- def set_output_embeddings(self, new_embeddings):
522
- self.lm_head = new_embeddings
523
-
524
  def prepare_inputs_for_generation(self, input_ids, past_key_values=None, inputs_embeds=None, **kwargs):
525
  if inputs_embeds is not None and past_key_values is None:
526
  model_inputs = {"inputs_embeds": inputs_embeds}
 
372
  def get_input_embeddings(self):
373
  return self.wte
374
 
 
 
 
375
  def forward(
376
  self,
377
  input_ids: Optional[torch.Tensor] = None,
 
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}