fix(framework): export unsloth lazily (#4992)

Signed-off-by: Aaron Pham <[email protected]>
bentoml · Sep 24, 2024 · f8a12b8 · f8a12b8
1 parent efce05b
commit f8a12b8
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"