remove Conv2d in white list in WOQ adaptor (#1274)

(cherry picked from commit 862c36c)
intel · Sep 27, 2023 · 1a6526f · 1a6526f
1 parent aff4131
commit 1a6526f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_compressor/adaptor/pytorch.py b/neural_compressor/adaptor/pytorch.py
@@ -4450,7 +4450,7 @@ def __init__(self, framework_specific_info):
             assert False, "Unsupported this device {}".format(self.device)
         self.query_handler = PyTorchQuery(local_config_file=os.path.join(os.path.dirname(__file__), query_config_file))
 
-        self.white_list = [torch.nn.Linear, torch.nn.Conv2d]
+        self.white_list = [torch.nn.Linear]
         # Contains parameters for algorithms such as AWQ, GPTQ, etc.
         self.recipes = framework_specific_info["recipes"]
         self.optype_statistics = None