Update modeling_gptj.py

modeling_gptj.py

@@ -40,22 +40,6 @@ from transformers.utils import (
 )
 from .configuration_gptj import GPTJConfig
 
-from mltools.dmx import DmxModel
-
-
-class DmxPreTrainedModel(PreTrainedModel):
-    @classmethod
-    def from_pretrained(cls, *args, **kwargs):
-        _model = super().from_pretrained(*args, **kwargs)
-        _model = DmxModel.from_torch(
-            _model, 
-            hf=True,
-            input_names=["input_ids"],  # TODO: no hard-coding!!!
-            concrete_args=None,
-        )
-        return _model
-
-
 logger = logging.get_logger(__name__)
 
 _CHECKPOINT_FOR_DOC = "hf-internal-testing/tiny-random-gptj"