From 3bec41f41a28c0d51111b49e0db712009cf0439f Mon Sep 17 00:00:00 2001
From: Roger Wang <136131678+ywang96@users.noreply.github.com>
Date: Tue, 2 Apr 2024 09:49:37 -0700
Subject: [PATCH] [Doc] Fix vLLMEngine Doc Page (#3791)

---
 docs/requirements-docs.txt | 4 +++-
 docs/source/conf.py        | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/docs/requirements-docs.txt b/docs/requirements-docs.txt
index 96749b93..0e76763a 100644
--- 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
\ No newline at end of file
+torch
+py-cpuinfo
+transformers
diff --git a/docs/source/conf.py b/docs/source/conf.py
index 61d8e55d..5619ea21 100644
--- 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",