Update pyproject.toml

nomadic-ml · Sep 11, 2024 · 07d25b1 · 07d25b1
1 parent 6300575
commit 07d25b1
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -43,15 +43,15 @@ pydantic = "^2.7.1"
 poetry-plugin-dotenv = "*"
 pandas = "^2.0.3"
 ray = "2.10.0"
-llama-index = "*"
+llama-index = "^0.10.68"
 llama-index-llms-sagemaker-endpoint = "^0.1.3"
-llama-index-llms-together = "*"
+llama-index-llms-together = "^0.1.3"
+llama-index-llms-vllm = "^0.1.9"
 flaml = "^1.2.0"
 matplotlib = "^3.7.5"
 optuna = "^3.6.1"
 seaborn = "^0.13.2"
 scipy = "^1.10.1"
-sentence-transformers = "^2.2.2"  
 
 [tool.poetry.group.dev.dependencies]
 requests = "*"