update colab install & fix colab path

2025-07-26 18:07:31 -07:00
parent 73927b68ef
commit c87c0fe662
4 changed files with 143 additions and 8 deletions
--- a/packages/leann-core/pyproject.toml
+++ b/packages/leann-core/pyproject.toml
@@ -33,6 +33,13 @@ dependencies = [
    "mlx-lm>=0.26.0; sys_platform == 'darwin'",
 ]

+[project.optional-dependencies]
+colab = [
+    "torch>=2.0.0,<3.0.0",  # 限制torch版本避免冲突
+    "transformers>=4.30.0,<5.0.0",  # 限制transformers版本
+    "accelerate>=0.20.0,<1.0.0",  # 限制accelerate版本
+]
+
 [project.scripts]
 leann = "leann.cli:main"