fix: rename incorrect access to model (#2)
Browse files- fix: rename incorrect access to model (414587a1760dcf6ccd622e92bd1d53e373ba6acf)
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
|
|
@@ -561,10 +561,10 @@ class StableLMEpochForCausalLM(StableLMEpochPreTrainedModel):
|
|
| 561 |
self.lm_head = new_embeddings
|
| 562 |
|
| 563 |
def get_decoder(self):
|
| 564 |
-
return self.
|
| 565 |
|
| 566 |
def set_decoder(self, decoder):
|
| 567 |
-
self.
|
| 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,
|