Spaces:
Runtime error
Runtime error
File size: 1,279 Bytes
7900c16 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
from tencentpretrain.embeddings.embedding import Embedding
from tencentpretrain.embeddings.dual_embedding import DualEmbedding
from tencentpretrain.embeddings.word_embedding import WordEmbedding
from tencentpretrain.embeddings.pos_embedding import PosEmbedding
from tencentpretrain.embeddings.seg_embedding import SegEmbedding
from tencentpretrain.embeddings.sinusoidalpos_embedding import SinusoidalposEmbedding
from tencentpretrain.embeddings.patch_embedding import PatchEmbedding
from tencentpretrain.embeddings.word_patch_embedding import WordPatchEmbedding
from tencentpretrain.embeddings.speech_embedding import SpeechEmbedding
from tencentpretrain.embeddings.masked_patch_embedding import MaskedPatchEmbedding
str2embedding = {"word": WordEmbedding, "pos": PosEmbedding, "seg": SegEmbedding,
"sinusoidalpos": SinusoidalposEmbedding, "dual": DualEmbedding,
"patch": PatchEmbedding, "word_patch": WordPatchEmbedding, "speech": SpeechEmbedding,
"masked_patch": MaskedPatchEmbedding}
__all__ = ["Embedding", "WordEmbedding", "PosEmbedding", "SegEmbedding", "SinusoidalposEmbedding",
"DualEmbedding", "PatchEmbedding", "WordPatchEmbedding", "SpeechEmbedding",
"MaskedPatchEmbedding", "str2embedding"]
|