fix: rename incorrect access to model

#2
Files changed (1) hide show
  1. modeling_stablelm_epoch.py +3 -3
modeling_stablelm_epoch.py CHANGED
@@ -549,7 +549,7 @@ class StableLMEpochForCausalLM(StableLMEpochPreTrainedModel):
549
  self.post_init()
550
 
551
  def get_input_embeddings(self):
552
- return self.transformer.embed_tokens
553
 
554
  def set_input_embeddings(self, value):
555
  self.model.embed_tokens = value
@@ -561,10 +561,10 @@ class StableLMEpochForCausalLM(StableLMEpochPreTrainedModel):
561
  self.lm_head = new_embeddings
562
 
563
  def get_decoder(self):
564
- return self.transformer
565
 
566
  def set_decoder(self, decoder):
567
- self.transformer = decoder
568
 
569
  def forward(
570
  self,
 
549
  self.post_init()
550
 
551
  def get_input_embeddings(self):
552
+ return self.model.embed_tokens
553
 
554
  def set_input_embeddings(self, value):
555
  self.model.embed_tokens = value
 
561
  self.lm_head = new_embeddings
562
 
563
  def get_decoder(self):
564
+ return self.model
565
 
566
  def set_decoder(self, decoder):
567
+ self.model = decoder
568
 
569
  def forward(
570
  self,