[Doc] Fix vLLMEngine Doc Page (#3791)

2024-04-02 09:49:37 -07:00
parent 0739b1947f
commit 3bec41f41a
2 changed files with 4 additions and 1 deletions
--- a/docs/requirements-docs.txt
+++ b/docs/requirements-docs.txt
@ -7,4 +7,6 @@ sphinx-argparse
 # packages to install to build the documentation
 pydantic
 -f https://download.pytorch.org/whl/cpu
-torch
+torch
+py-cpuinfo
+transformers
--- a/docs/source/conf.py
+++ b/docs/source/conf.py
@ -75,6 +75,7 @@ html_theme_options = {

 # Mock out external dependencies here.
 autodoc_mock_imports = [
+    "cpuinfo",
    "torch",
    "transformers",
    "psutil",