__init__.py

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