Update configuration_opt.py
Browse files- configuration_opt.py +2 -1
configuration_opt.py
CHANGED
|
@@ -20,7 +20,7 @@ from transformers.utils import logging
|
|
| 20 |
|
| 21 |
logger = logging.get_logger(__name__)
|
| 22 |
|
| 23 |
-
|
| 24 |
|
| 25 |
|
| 26 |
class OPTConfig(PretrainedConfig):
|
|
@@ -146,3 +146,4 @@ class OPTConfig(PretrainedConfig):
|
|
| 146 |
# see https://github.com/facebookresearch/metaseq/pull/164
|
| 147 |
self._remove_final_layer_norm = _remove_final_layer_norm
|
| 148 |
self.attn_implementation = attn_implementation
|
|
|
|
|
|
| 20 |
|
| 21 |
logger = logging.get_logger(__name__)
|
| 22 |
|
| 23 |
+
|
| 24 |
|
| 25 |
|
| 26 |
class OPTConfig(PretrainedConfig):
|
|
|
|
| 146 |
# see https://github.com/facebookresearch/metaseq/pull/164
|
| 147 |
self._remove_final_layer_norm = _remove_final_layer_norm
|
| 148 |
self.attn_implementation = attn_implementation
|
| 149 |
+
OPTConfig.register_for_auto_class()
|