robinzixuan commited on
Commit
ceadea2
1 Parent(s): 5763f2b

Update configuration_opt.py

Browse files
Files changed (1) hide show
  1. 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
- OPTConfig.register_for_auto_class()
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()