fix(import): set unsloth export lazily

Signed-off-by: Aaron Pham <[email protected]>
bentoml · Sep 24, 2024 · a24d709 · a24d709
1 parent efce05b
commit a24d709
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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"