ThingsAI commited on
Commit
e0d9139
·
verified ·
1 Parent(s): cd6716a

fix: tie_weights(**kwargs) per compatibilità transformers 4.40+

Browse files
Files changed (1) hide show
  1. modeling_quark.py +1 -1
modeling_quark.py CHANGED
@@ -157,7 +157,7 @@ class QuarkForCausalLM(QuarkPreTrainedModel):
157
  def set_output_embeddings(self, value):
158
  self.lm_head = value
159
 
160
- def tie_weights(self):
161
  """HF chiama questo metodo dopo il caricamento dei pesi."""
162
  if self.config.tie_word_embeddings:
163
  self.lm_head.weight = self.embed_tokens.weight
 
157
  def set_output_embeddings(self, value):
158
  self.lm_head = value
159
 
160
+ def tie_weights(self, **kwargs):
161
  """HF chiama questo metodo dopo il caricamento dei pesi."""
162
  if self.config.tie_word_embeddings:
163
  self.lm_head.weight = self.embed_tokens.weight