Skip to content

Commit 790fe31

Browse files
kylesayrsdsikka
andauthored
[Tests] Workaround qwen_2_5_vl (#1894)
## Purpose ## * Fix qwen_2_5_vl tests Signed-off-by: Kyle Sayers <[email protected]> Co-authored-by: Dipika Sikka <[email protected]>
1 parent ec298db commit 790fe31

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

src/llmcompressor/pipelines/sequential/helpers.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -303,10 +303,12 @@ def topological_partition(graph: GraphModule, targets: Set[Module]) -> List[List
303303
if user in partitions[index]:
304304
user_partitions.append(index)
305305
break
306-
partition_index = min(user_partitions)
307-
partitions[partition_index].insert(0, node)
308306

309-
assert set().union(*partitions) == set(graph.graph.nodes)
307+
# workaround
308+
if len(user_partitions):
309+
partition_index = min(user_partitions)
310+
partitions[partition_index].insert(0, node)
311+
310312
return partitions
311313

312314

0 commit comments

Comments
 (0)