Skip to content

Commit 90fb431

Browse files
fix layer match (#1873)
Signed-off-by: Kaihui-intel <kaihui.tang@intel.com> Co-authored-by: Sun, Xuehao <xuehao.sun@intel.com>
1 parent f4eb660 commit 90fb431

File tree

1 file changed

+1
-1
lines changed
  • neural_compressor/torch/algorithms/weight_only

1 file changed

+1
-1
lines changed

neural_compressor/torch/algorithms/weight_only/gptq.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -118,7 +118,7 @@ def find_layers(module, layers=SUPPORTED_LAYERS, name=""):
118118
return {name: module}
119119
else:
120120
# use string type to find name:
121-
if type(module).__name__ in ["Linear"]:
121+
if isinstance(module, tuple(layers)):
122122
return {name: module}
123123
else:
124124
pass

0 commit comments

Comments
 (0)