Update modeling_codefuse_cge_small.py

modeling_codefuse_cge_small.py

@@ -79,6 +79,7 @@ import torch.nn as nn
 import torch.nn.functional as F
 import math
 import re
+from CodeFuseCGESmallConfig import CodeFuseCGESmallConfig
 
 
 class MAB_POST(nn.Module):
@@ -142,7 +143,7 @@ class PMA(nn.Module):
 
 
 class CodeFuse_CGE_Small(PreTrainedModel):
-
+    config_class = CodeFuseCGESmallConfig
     def __init__(self, config):
         super().__init__(config)
         self.plm_model = Phi3ForCausalLM(config)