separate requirements for speculative training and benchmarking (#192)

Author: liusong1222

requirements/requirements_benchmark.txt

@@ -0,0 +1 @@
+vllm>=0.11.0

requirements/requirements_speculative.txt

@@ -5,5 +5,4 @@ ray
 referencing
 jsonschema_specifications
 deepspeed
-wandb
-vllm>=0.11.0
+wandb==0.22.1

setup.py

@@ -52,13 +52,16 @@ def get_requirements(filename):
             get_requirements("requirements/requirements_speculative.txt")
             + get_requirements("requirements/requirements_diffusion.txt")
             + get_requirements("requirements/requirements_multimodal.txt")
+            + get_requirements("requirements/requirements_benchmark.txt")
         ),
         # Install speculative sampling functionality: pip install angelslim[speculative]
         "speculative": get_requirements("requirements/requirements_speculative.txt"),
         # Install Diffusion functionality: pip install angelslim[diffusion]
         "diffusion": get_requirements("requirements/requirements_diffusion.txt"),
         # Install multimodal functionality: pip install angelslim[multimodal]
         "multimodal": get_requirements("requirements/requirements_multimodal.txt"),
+        # Install benchmark functionality: pip install angelslim[benchmark]
+        "benchmark": get_requirements("requirements/requirements_benchmark.txt"),
     },
     packages=find_packages(),
     python_requires=">=3.0",