TanishqV03 commited on
Commit
76da2e9
·
verified ·
1 Parent(s): cdcb726

Fix: add all_tied_weights_keys property

Browse files
Files changed (1) hide show
  1. modeling_easysell.py +4 -0
modeling_easysell.py CHANGED
@@ -115,6 +115,10 @@ class EasySellForCausalLM(PreTrainedModel, GenerationMixin):
115
  n_embd=config.n_embd, dropout=config.dropout)
116
  self.model = GPT(gpt_cfg)
117
 
 
 
 
 
118
  def forward(self, input_ids, labels=None, **kwargs):
119
  logits, loss = self.model(input_ids, targets=labels)
120
  from transformers.modeling_outputs import CausalLMOutput
 
115
  n_embd=config.n_embd, dropout=config.dropout)
116
  self.model = GPT(gpt_cfg)
117
 
118
+ @property
119
+ def all_tied_weights_keys(self):
120
+ return {}
121
+
122
  def forward(self, input_ids, labels=None, **kwargs):
123
  logits, loss = self.model(input_ids, targets=labels)
124
  from transformers.modeling_outputs import CausalLMOutput