Skip to content

Commit afeebf0

Browse files
committed
fix: skip output node in FX graph cleanup to fix erase_node crash (Granite4 GPTQ)
Signed-off-by: gillesturpin <turpingilles@orange.fr>
1 parent e6fdd06 commit afeebf0

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/llmcompressor/pipelines/sequential/transformers_helpers.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1478,6 +1478,8 @@ def to_meta(value):
14781478
to_delete = collections.OrderedDict()
14791479
while to_visit:
14801480
n = to_visit.pop(0)
1481+
if n in to_delete:
1482+
to_delete.move_to_end(n)
14811483
to_delete[n] = None
14821484
to_visit += list(n.users.keys())
14831485

0 commit comments

Comments
 (0)