from similarity.config import BERTMultiAttentionConfig
from similarity.model import BERTMultiAttentionModel, MultiHeadAttention