from __future__ import annotations

from .CSRSparsity import CSRSparsity
from .TopKActivation import TopKActivation

__all__ = ["CSRSparsity", "TopKActivation"]
# TODO : Add in models the possibility to have the MLM head(for splade)

from __future__ import annotations

from .CSRSparsity import CSRSparsity

__all__ = ["CSRSparsity"]
