fix: Missing get_input_embeddings / set_input_embeddings on NomicBertModel
#3
by Pringled - opened
modeling_hf_nomic_bert.py
CHANGED
|
@@ -1050,6 +1050,12 @@ class NomicBertModel(NomicBertPreTrainedModel):
|
|
| 1050 |
|
| 1051 |
self.apply(partial(_init_weights, initializer_range=config.initializer_range))
|
| 1052 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1053 |
def forward(
|
| 1054 |
self,
|
| 1055 |
input_ids,
|
|
|
|
| 1050 |
|
| 1051 |
self.apply(partial(_init_weights, initializer_range=config.initializer_range))
|
| 1052 |
|
| 1053 |
+
def get_input_embeddings(self):
|
| 1054 |
+
return self.embeddings.word_embeddings
|
| 1055 |
+
|
| 1056 |
+
def set_input_embeddings(self, value):
|
| 1057 |
+
self.embeddings.word_embeddings = value
|
| 1058 |
+
|
| 1059 |
def forward(
|
| 1060 |
self,
|
| 1061 |
input_ids,
|