From ec86041e1b3d9a602580ad4f636114dbdfe9be6f Mon Sep 17 00:00:00 2001 From: chensuyue Date: Fri, 30 Aug 2024 17:51:57 +0800 Subject: [PATCH] update 3x pt binary build Signed-off-by: chensuyue Signed-off-by: chensuyue --- setup.py | 2 ++ 1 file changed, 2 insertions(+) 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"),