Make vllm optional

pyproject.toml

@@ -20,6 +20,10 @@ classifiers = [
 ]
 dynamic = ["dependencies"]
 
+[project.optional-dependencies]
+vllm = ["vllm"]
+tests = ["pytest", "pytest-mock"]
+
 [project.urls]
 "Homepage" = "https://github.com/facebookresearch/llama-recipes/"
 "Bug Tracker" = "https://github.com/facebookresearch/llama-recipes/issues"

requirements.txt

@@ -12,7 +12,4 @@ transformers>=4.31.0
 sentencepiece
 py7zr
 scipy
-optimum
-vllm
-pytest
-pytest-mock
+optimum