diff --git a/wenet/utils/class_utils.py b/wenet/utils/class_utils.py index 3bda4db20..547117e9b 100644 --- a/wenet/utils/class_utils.py +++ b/wenet/utils/class_utils.py @@ -49,6 +49,7 @@ } WENET_EMB_CLASSES = { + "embed": PositionalEncoding, "abs_pos": PositionalEncoding, "rel_pos": RelPositionalEncoding, "no_pos": NoPositionalEncoding,