robinzixuan
commited on
Commit
•
ceadea2
1
Parent(s):
5763f2b
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()
|