Update modeling_codefuse_cge_large.py

modeling_codefuse_cge_large.py

@@ -81,7 +81,7 @@ import torch.nn as nn
 import torch.nn.functional as F
 import math
 import re
-
+from .configuration_codefuse_cge_large import CodeFuseCGELargeConfig
 
 class MAB_POST(nn.Module):
     def __init__(self, dim_Q, dim_K, dim_V, num_heads, ln=False):
@@ -144,7 +144,7 @@ class PMA(nn.Module):
 
 
 class CodeFuse_CGE_Large(PreTrainedModel):
-
+    config_class = CodeFuseCGELargeConfig
     def __init__(self, config):
         super().__init__(config)
         self.plm_model = Qwen2ForCausalLM(config)