Update mlm.py

mlm.py

@@ -10,9 +10,9 @@ from transformers.modeling_outputs import (
     SequenceClassifierOutput,
 )
 
-from energybert.hopfield import HopfieldLayer
-from energybert.hf_configuration import BertEnergyConfig
-from energybert.positional import PositionalEncoding
+from .hopfield import HopfieldLayer
+from .hf_configuration import BertEnergyConfig
+from .positional import PositionalEncoding
 
 
 class EnergyLMHead(nn.Module):