Update modeling_opt.py

modeling_opt.py

@@ -53,6 +53,10 @@ logger = logging.get_logger(__name__)
 _CHECKPOINT_FOR_DOC = "facebook/opt-350m"
 _CONFIG_FOR_DOC = "OPTConfig"
 
+
+
+
+
 # Base model docstring
 _EXPECTED_OUTPUT_SHAPE = [1, 8, 1024]
 
@@ -76,6 +80,9 @@ def _get_unpad_data(attention_mask):
     )
 
 
+OPTOutEffHop.register_for_auto_class("AutoModel")
+
+    
 class OPTLearnedPositionalEmbedding(nn.Embedding):
     """
     This module learns positional embeddings up to a fixed maximum size.