fix: rename incorrect access to model
Browse files
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.
|
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
|