feat: unsloth integrations

Signed-off-by: Aaron Pham <[email protected]>

.gitignore

@@ -167,7 +167,3 @@ mlruns/
 .pdm-python
 .python-version
 .pdm-build/
-
-# from training scripts
-model
-outputs

.pre-commit-config.yaml

@@ -2,7 +2,7 @@ ci:
   autoupdate_schedule: monthly
   autofix_commit_msg: "ci: auto fixes from pre-commit.ci\n\nFor more information, see https://pre-commit.ci"
   autoupdate_commit_msg: 'ci: pre-commit autoupdate [skip ci]'
-  skip: # exceeds tier max size
+  skip:  # exceeds tier max size
     - buf-format
     - buf-lint
 exclude: '(.*\.(css|js|svg))|(.*/(snippets|grpc|proto)/.*)$'
@@ -13,9 +13,7 @@ repos:
       - id: ruff
         args: [--fix, --exit-non-zero-on-fix, --show-fixes]
         types_or: [python, pyi]
-        exclude: ^src/_bentoml_impl/frameworks/unsloth/train\.py$
       - id: ruff-format
-        exclude: ^src/_bentoml_impl/frameworks/unsloth/train\.py$
         types_or: [python, pyi]
         files: '(src|tests|docs|examples|typings)/'
   - repo: https://github.com/pre-commit/pre-commit-hooks

pyproject.toml

@@ -93,11 +93,7 @@ io = [
 io-image = ["Pillow"]
 io-pandas = ["pandas>=1", "pyarrow"]
 triton = ["tritonclient>=2.29.0", "tritonclient[all]; sys_platform != 'darwin'"]
-unsloth = [
-    "unsloth[huggingface] @ git+https://github.com/bentoml/unsloth.git@main",
-    "vllm>=0.5.5",
-    "fastapi"
-]
+unsloth = ["bentoml-unsloth"]
 grpc = [
     "protobuf",
     "grpcio",
@@ -243,11 +239,6 @@ testpaths = ["tests"]
 line-length = 88
 target-version = "py310"
 
-[tool.ruff.format]
-exclude = [
-    "src/_bentoml_impl/frameworks/unsloth/train.py",
-]
-
 [tool.ruff.lint]
 # We ignore E501 (line too long) here because we keep user-visible strings on one line.
 ignore = ["E501"]
@@ -260,7 +251,6 @@ exclude = [
     "src/bentoml/_internal/external_typing",
     "src/bentoml/grpc/v1alpha1",
     "src/bentoml/grpc/v1",
-    "src/_bentoml_impl/frameworks/unsloth",
     "tests/proto",
 ]