Add bias for fx_model (#1447)

Signed-off-by: Tang, Kaihui <kaihui.tang@intel.com> Co-authored-by: chen, suyue <suyue.chen@intel.com>
intel · Dec 12, 2023 · 7f53d16 · 7f53d16
1 parent c741b89
commit 7f53d16
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/neural_compressor/adaptor/torch_utils/util.py b/neural_compressor/adaptor/torch_utils/util.py
@@ -184,6 +184,11 @@ def append_attr(fx_model, model, fx_white_list=[]):
             fx_model.weight = fx_model.module.weight()
         else:
             fx_model.weight = fx_model.module.weight
+        if hasattr(fx_model.module, "bias"):
+            if not isinstance(fx_model.module.bias, torch.Tensor) and fx_model.module.bias is not None:
+                fx_model.bias = fx_model.module.bias()
+            else:
+                fx_model.bias = fx_model.module.bias
     for i in org_attr:
         if (
             type(model) in fx_white_list