oweller2
commited on
Commit
•
c87aa93
1
Parent(s):
2ee7515
modeling
Browse files- modeling_flexbert.py +1 -0
modeling_flexbert.py
CHANGED
|
@@ -116,6 +116,7 @@ from .layers import (
|
|
| 116 |
from .mlp import FlexBertGLU, FlexBertMLP, FlexBertParallelGLU
|
| 117 |
from .normalization import get_norm_layer
|
| 118 |
from .padding import pad_input, unpad_input
|
|
|
|
| 119 |
|
| 120 |
# TODO: This is not used here, but this is so these files are copied when saving the model in ST/PyLate
|
| 121 |
from .utils import StrEnum
|
|
|
|
| 116 |
from .mlp import FlexBertGLU, FlexBertMLP, FlexBertParallelGLU
|
| 117 |
from .normalization import get_norm_layer
|
| 118 |
from .padding import pad_input, unpad_input
|
| 119 |
+
from .bert_layers.loss import get_loss_fn
|
| 120 |
|
| 121 |
# TODO: This is not used here, but this is so these files are copied when saving the model in ST/PyLate
|
| 122 |
from .utils import StrEnum
|