diff --git a/setup.py b/setup.py index 2eddbbebd26..09f9d5939e1 100644 --- a/setup.py +++ b/setup.py @@ -68,6 +68,8 @@ def get_build_version(): "neural_compressor.common.*", "neural_compressor.torch", "neural_compressor.torch.*", + "neural_compressor.transformers", + "neural_compressor.transformers.*", ], ), "install_requires": fetch_requirements("requirements_pt.txt"),