fix import err

modeling_muddpythia.py

@@ -9,10 +9,11 @@ from torch import Tensor
 from torch.nn import functional as F
 from torch.utils.checkpoint import checkpoint
 
-try:
-    from .configuration_muddpythia import MUDDPythiaConfig
-except:
-    from configuration_muddpythia import MUDDPythiaConfig
+from .configuration_muddpythia import MUDDPythiaConfig
+#try:
+#    from .configuration_muddpythia import MUDDPythiaConfig
+#except:
+#    from configuration_muddpythia import MUDDPythiaConfig
 
 from transformers.modeling_utils import PreTrainedModel
 
@@ -451,4 +452,4 @@ def match_weight_muddpythia(model, w, strict=False, pythia=True):
             v = v+1
         state_dict[k] = torch.tensor(v)
     model.load_state_dict(state_dict, strict=strict)
-    return model
+    return model