utils/__init__.py

from . import helpers
from . import losses
from .losses import MLMLoss
from .helpers import create_masked_input, create_multimodal_model, get_max, get_token_embeddings