Update modeling_gptj.py

modeling_gptj.py

@@ -363,7 +363,7 @@ class GPTJBlock(nn.Module):
         return outputs  # hidden_states, present, (attentions)
 
 
-class GPTJPreTrainedModel(DmxPreTrainedModel):
+class GPTJPreTrainedModel(PreTrainedModel):
     """
     An abstract class to handle weights initialization and a simple interface for downloading and loading pretrained
     models.