File size: 350 Bytes
473c3a0
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
from distiller.model2vec.utils import get_package_extras, importable

_REQUIRED_EXTRA = "distill"

for extra_dependency in get_package_extras("model2vec", _REQUIRED_EXTRA):
    importable(extra_dependency, _REQUIRED_EXTRA)

from distiller.model2vec.distill.distillation import distill, distill_from_model

__all__ = ["distill", "distill_from_model"]