Update modeling_opt.py

modeling_opt.py

@@ -554,7 +554,7 @@ class OPTDecoderLayer(nn.Module):
         super().__init__()
         self.embed_dim = config.hidden_size
 
-        self.self_attn = OPT_ATTENTION_CLASSES[config._attn_implementation](
+        self.self_attn = OPTAttention(
             config=config, is_decoder=True)
         print(self.self_attn)
         print(config._attn_implementation)