Update modeling_codefuse_cge_small.py
Browse files
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 .
|
| 83 |
|
| 84 |
|
| 85 |
class MAB_POST(nn.Module):
|
|
|
|
| 79 |
import torch.nn.functional as F
|
| 80 |
import math
|
| 81 |
import re
|
| 82 |
+
from .configuration_codefuse_cge_small import CodeFuseCGESmallConfig
|
| 83 |
|
| 84 |
|
| 85 |
class MAB_POST(nn.Module):
|