diff --git a/docs/requirements-docs.txt b/docs/requirements-docs.txt index 96749b9327d7..0e76763a87b7 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 61d8e55d2cc6..5619ea219193 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",