diff --git a/docs/source/conf.py b/docs/source/conf.py index 5a45c6f9..61d24e16 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -72,8 +72,15 @@ html_theme_options = { # Mock out external dependencies here. autodoc_mock_imports = [ - "torch", "transformers", "psutil", "prometheus_client", "sentencepiece", - "vllm.cuda_utils", "vllm._C" + "torch", + "transformers", + "psutil", + "prometheus_client", + "sentencepiece", + "vllm.cuda_utils", + "vllm._C", + "numpy", + "tqdm", ] for mock_target in autodoc_mock_imports: