yoshimitsu-imazu
commited on
Commit
•
e7d2ebd
1
Parent(s):
e98f084
fix: rename incorrect access to model (#2)
Browse files- fix: rename incorrect access to model (d8f00afe1e260e53ffcf1fd75e8b742e28f10cd9)
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,
|