Update modeling.py
Browse files- modeling.py +4 -1
modeling.py
CHANGED
|
@@ -154,4 +154,7 @@ class CustomModel(PreTrainedModel):
|
|
| 154 |
return CausalLMOutput(logits=logits)
|
| 155 |
|
| 156 |
def get_input_embeddings(self):
|
| 157 |
-
return self.model.embed
|
|
|
|
|
|
|
|
|
|
|
|
| 154 |
return CausalLMOutput(logits=logits)
|
| 155 |
|
| 156 |
def get_input_embeddings(self):
|
| 157 |
+
return self.model.embed
|
| 158 |
+
|
| 159 |
+
def set_input_embeddings(self, x):
|
| 160 |
+
self.model.embed = x
|