Text Generation
Transformers
Safetensors
English
stablelm
causal-lm
Eval Results
Inference Endpoints
8 papers
winglian commited on
Commit
e5d02b2
1 Parent(s): a4750ac

fix get_input_embdeddings

Browse files

fixes
` File "/root/.cache/huggingface/modules/transformers_modules/stabilityai/stablelm-3b-4e1t/a4750ace0db6f08d7bbba0aa52a585f231ea3cde/modeling_stablelm_epoch.py", line 552, in get_input_embeddings
return self.transformer.embed_tokens`

Files changed (1) hide show
  1. modeling_stablelm_epoch.py +1 -1
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
 
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