codefuse-admin commited on
Commit
8f1a250
·
verified ·
1 Parent(s): fe168dd

Update modeling_codefuse_cge_small.py

Browse files
Files changed (1) hide show
  1. modeling_codefuse_cge_small.py +1 -1
modeling_codefuse_cge_small.py CHANGED
@@ -79,7 +79,7 @@ import torch.nn as nn
79
  import torch.nn.functional as F
80
  import math
81
  import re
82
- from CodeFuseCGESmallConfig import CodeFuseCGESmallConfig
83
 
84
 
85
  class MAB_POST(nn.Module):
 
79
  import torch.nn.functional as F
80
  import math
81
  import re
82
+ from .CodeFuseCGESmallConfig import CodeFuseCGESmallConfig
83
 
84
 
85
  class MAB_POST(nn.Module):