sarthak1
/

codemalt

Feature Extraction

sentence-transformers

code-embeddings

static-embeddings

Model card Files Files and versions

codemalt / src /distiller /model2vec /inference /__init__.py

Sarthak

chore: moved model2vec as in internal package

473c3a0 about 1 year ago

History Blame Contribute Delete

395 Bytes

	from distiller.model2vec.utils import get_package_extras, importable

	_REQUIRED_EXTRA = "inference"

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

	from distiller.model2vec.inference.model import StaticModelPipeline, evaluate_single_or_multi_label

	__all__ = ["StaticModelPipeline", "evaluate_single_or_multi_label"]