diff --git a/pyproject.toml b/pyproject.toml index 1567b14..faef3bc 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,7 +19,7 @@ license = {text = "Apache 2"} dependencies = [ "argilla>=2.4.0,<3.0.0", - "distilabel[argilla,hf-inference-endpoints,hf-transformers,instructor,llama-cpp,ollama,openai,outlines,vllm] @ git+https://github.com/argilla-io/distilabel.git@develop", + "distilabel[argilla,hf-inference-endpoints,hf-transformers,instructor,llama-cpp,ollama,openai,outlines,vllm,vision]>=1.5.0,<2.00", "gradio[oauth]>=5.4.0,<6.0.0", "gradio-huggingfacehub-search>=0.0.12,<1.0.0", "model2vec>=0.2.4,<1.0.0", @@ -27,6 +27,7 @@ dependencies = [ "sentence-transformers>=3.2.0,<4.0.0", "transformers>=4.44.2,<5.0.0", "unstructured[md,pdf,docx]>=0.16.0,<1.0.0", + "setuptools", ] [build-system]