Fix: add all_tied_weights_keys property
Browse files- 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
|