Update modeling_opt.py
Browse files- modeling_opt.py +3 -1
modeling_opt.py
CHANGED
|
@@ -80,7 +80,7 @@ def _get_unpad_data(attention_mask):
|
|
| 80 |
)
|
| 81 |
|
| 82 |
|
| 83 |
-
|
| 84 |
|
| 85 |
|
| 86 |
class OPTLearnedPositionalEmbedding(nn.Embedding):
|
|
@@ -1726,3 +1726,5 @@ class OPTForQuestionAnswering(OPTPreTrainedModel):
|
|
| 1726 |
|
| 1727 |
def set_input_embeddings(self, value):
|
| 1728 |
self.model.decoder.embed_tokens = value
|
|
|
|
|
|
|
|
|
| 80 |
)
|
| 81 |
|
| 82 |
|
| 83 |
+
|
| 84 |
|
| 85 |
|
| 86 |
class OPTLearnedPositionalEmbedding(nn.Embedding):
|
|
|
|
| 1726 |
|
| 1727 |
def set_input_embeddings(self, value):
|
| 1728 |
self.model.decoder.embed_tokens = value
|
| 1729 |
+
|
| 1730 |
+
OPTModel.register_for_auto_class("AutoModel")
|