diff --git a/src/bentoml/__init__.py b/src/bentoml/__init__.py index ae281ea13a2..41924ad1987 100644 --- a/src/bentoml/__init__.py +++ b/src/bentoml/__init__.py @@ -132,6 +132,9 @@ lightgbm = _LazyLoader( "bentoml.lightgbm", globals(), "_bentoml_impl.frameworks.lightgbm" ) + unsloth = _LazyLoader( + "bentoml.unsloth", globals(), "_bentoml_impl.frameworks.unsloth" + ) mlflow = _LazyLoader("bentoml.mlflow", globals(), "_bentoml_impl.frameworks.mlflow") detectron = _LazyLoader( "bentoml.detectron", globals(), "bentoml._internal.frameworks.detectron"