stabilityai
/

stablelm-3b-4e1t

Text Generation

Inference Endpoints

Model card Files Files and versions Community

fix get_input_embdeddings

#3

by winglian - opened Oct 3, 2023

base: refs/heads/main

←

from: refs/pr/3

Discussion Files changed

Files changed (1) hide show

modeling_stablelm_epoch.py +1 -1

modeling_stablelm_epoch.py CHANGED Viewed

@@ -549,7 +549,7 @@ class StableLMEpochForCausalLM(StableLMEpochPreTrainedModel):
         self.post_init()
     def get_input_embeddings(self):
-        return self.transformer.embed_tokens
     def set_input_embeddings(self, value):
         self.model.embed_tokens = value

         self.post_init()
     def get_input_embeddings(self):
+        return self.model.embed_tokens
     def set_input_embeddings(self, value):
         self.model.embed_tokens = value