Text Generation
Transformers
Safetensors
English
stablelm
causal-lm
Eval Results
Inference Endpoints
8 papers
jon-tow commited on
Commit
c6554ba
1 Parent(s): 5951257

fix: remove `StableLMEpochForCausalLM.transformer` refs

Browse files
Files changed (1) hide show
  1. modeling_stablelm_epoch.py +2 -2
modeling_stablelm_epoch.py CHANGED
@@ -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,
 
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,