From a24d70953f297f90011a41dfee5e944f2830e19e Mon Sep 17 00:00:00 2001 From: Aaron Pham Date: Mon, 23 Sep 2024 20:19:37 -0400 Subject: [PATCH] fix(import): set unsloth export lazily Signed-off-by: Aaron Pham --- src/bentoml/__init__.py | 3 +++ 1 file changed, 3 insertions(+) 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"